jax
8b02884c
- [Mosaic:TPU] Fix trunc infer rule after cl/708011538
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
1 year ago
[Mosaic:TPU] Fix trunc infer rule after cl/708011538 For (1, 128) tiling 32-bit input, it assigns (1, 128) tiling at output, which can be invalid (e.g. it should be (1, 256) for bf16) PiperOrigin-RevId: 708112341
References
#25624 - [Mosaic:TPU] Fix trunc infer rule after cl/708011538
#190 - CI: 12/23/24 upstream sync
Author
tlongeri
Committer
Google-ML-Automation
Parents
64d3871e
Loading