jax
ee3d8843
- [Mosaic TPU] Support 16-bit `arith::CmpI` and 1d tiling subelement mask on TPU v4 and v5.
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
69 days ago
[Mosaic TPU] Support 16-bit `arith::CmpI` and 1d tiling subelement mask on TPU v4 and v5. PiperOrigin-RevId: 892539421
References
#36306 - [Mosaic TPU] Support 16-bit `arith::CmpI` and 1d tiling subelement mask on TPU v4 and v5.
Author
yueshengys
Committer
Google-ML-Automation
Parents
dfc7e729
Loading