jax
bf618a83
- [Mosaic:TPU] 16-bit tpu.iota
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
332 days ago
[Mosaic:TPU] 16-bit tpu.iota To support expressing iota over half-sublanes with a vreg shape of 8x128x2, for tpu.iota we now specify multiple dimensions. PiperOrigin-RevId: 775574142
References
#28987 - [Mosaic:TPU] 16-bit tpu.iota
Author
tlongeri
Committer
Google-ML-Automation
Parents
f54f6801
Loading