jax
65e83bd8 - [Mosaic:TPU][Relayouts] (1, 128 * packing) <-> (n * packing, 128) retilings for n > 1

Commit

335 days ago

[Mosaic:TPU][Relayouts] (1, 128 * packing) <-> (n * packing, 128) retilings for n > 1 This is done simply by chaining existing relayouts. PiperOrigin-RevId: 777354672

References

#29732 - [Mosaic:TPU][Relayouts] (1, 128 * packing) <-> (n * packing, 128) retilings for n > 1

Author

tlongeri

Committer

Google-ML-Automation

Parents

172e3cd3

jax 65e83bd8 - [Mosaic:TPU][Relayouts] (1, 128 * packing) <-> (n * packing, 128) retilings for n > 1

jax
65e83bd8 - [Mosaic:TPU][Relayouts] (1, 128 * packing) <-> (n * packing, 128) retilings for n > 1