jax
d44b16cf - [XLA:Mosaic] Generalize (8,128) -> (8 * packing,128) retiling for packed type.

Commit

1 year ago

[XLA:Mosaic] Generalize (8,128) -> (8 * packing,128) retiling for packed type. PiperOrigin-RevId: 625816937

Author

bythew3i

Committer

a-googler

Parents