flax
17fe4a73 - allow in/out_axes when graph_updates=False

Commit

13 hours ago

allow in/out_axes when graph_updates=False PiperOrigin-RevId: 881145617

References

#5323 - allow in/out_axes when graph_updates=False

Author

Cristian Garcia

Committer

a-googler

a-googler

Parents

Loading