PR #3947 [nnx] improve vmap axis size detection

[nnx] improve vmap axis size detection #3947

copybara-service merged 1 commit into main from nnx-vmap-improve-axes-size-detection

[nnx] improve vmap axis size detection

90d656a8

chiamp requested a review from

chiamp 1 year ago

chiamp approved these changes on 2024-05-28

cgarciae added pull ready

copybara-service merged 2cfd1747 into main 1 year ago

copybara-service deleted the nnx-vmap-improve-axes-size-detection branch 1 year ago

Reviewers

chiamp

Assignees

No one assigned

Labels

pull ready

Milestone

No milestone