jax
85193bdc
- [NFC][Mosaic:TPU] Minor refactor of changeTiling to make chaining packs/unpacks and relayouts easier
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
254 days ago
[NFC][Mosaic:TPU] Minor refactor of changeTiling to make chaining packs/unpacks and relayouts easier Make the src layout mutable and get rid of src_offsets and src_tiling variables (get these directly from src) PiperOrigin-RevId: 775884959
References
#29730 - [NFC][Mosaic:TPU] Minor refactor of changeTiling to make chaining packs/unpacks and relayouts easier
Author
tlongeri
Committer
Google-ML-Automation
Parents
45008fab
Loading