flax
437bb82b - fix nn.vmap's _broadcast_prefix_tree

Commit

58 days ago

fix nn.vmap's _broadcast_prefix_tree PiperOrigin-RevId: 875873975

References

#5282 - fix nn.vmap's _broadcast_prefix_tree

Author

Cristian Garcia

Committer

a-googler

a-googler

Parents

Loading