jax
9752ba97 - [Mosaic] Allow arbitrary broadcasts on non-tiled dimensions

Commit

2 years ago

[Mosaic] Allow arbitrary broadcasts on non-tiled dimensions PiperOrigin-RevId: 577145133

Author

apaszke

Committer

a-googler

Parents