jax
68b08bd3 - [Mosaic GPU] Validate `source_dimensions` in `broadcast_in_dim`.

Commit

15 days ago

[Mosaic GPU] Validate `source_dimensions` in `broadcast_in_dim`. PiperOrigin-RevId: 879639117

References

Author

allanrenucci

Committer

Google-ML-Automation

Parents