jax
8b958536 - [Mosaic] Add relayout for (1, 128 * packing) -> (packing, 128).

Commit

1 year ago

[Mosaic] Add relayout for (1, 128 * packing) -> (packing, 128). PiperOrigin-RevId: 637951690

References

Author

tlongeri

Committer

a-googler

Parents