Skip to content

Have _graph_flatten respect nnx.data declarations (extension of #5140)#5159

Merged
chapman20j merged 2 commits intogoogle:mainfrom
samanklesaria:pr/5140
Jan 5, 2026
Merged

Have _graph_flatten respect nnx.data declarations (extension of #5140)#5159
chapman20j merged 2 commits intogoogle:mainfrom
samanklesaria:pr/5140

Conversation

@samanklesaria
Copy link
Collaborator

@samanklesaria samanklesaria commented Dec 29, 2025

Fixes #5128

This is just PR #5140 from @cgarciae, but fixes its type error.

@chapman20j chapman20j merged commit 4cc1d1b into google:main Jan 5, 2026
17 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

nnx.data is lost during nnx.eval_shape

3 participants