PR #5141 Handle pure bodies in nnx.fori_loop

Handle pure bodies in nnx.fori_loop #5141

copybara-service merged 2 commits into google:main from samanklesaria:issues/5113

Handle pure bodies in nnx.fori_loop

7255c612

samanklesaria requested a review from

cgarciae 112 days ago

samanklesaria commented on 2025-12-17

samanklesaria requested a review from

vfdev-5 93 days ago

IvyZX requested changes on 2026-01-12

Use eval_shape for initial call to body

573cd25d

samanklesaria requested a review from

IvyZX 86 days ago

IvyZX approved these changes on 2026-01-13

IvyZX added pull ready

copybara-service merged b4252f89 into main 85 days ago

Reviewers

IvyZX

cgarciae

vfdev-5

Assignees

No one assigned

Labels

pull ready

Milestone

No milestone