jax
e482f909 - [Mosaic:TPU] Fix bug in `rotateVregRows`

Commit
250 days ago
[Mosaic:TPU] Fix bug in `rotateVregRows` Shift ops expect RHS to have the same type as RHS, but a scalar was being used for RHS. PiperOrigin-RevId: 747934619
Author
Committer
Parents
Loading