jax
a2f22cc1 - [Mosaic GPU] Adding a primitive to load from memrefs with a specified layout.

Commit

328 days ago

[Mosaic GPU] Adding a primitive to load from memrefs *with* a specified layout. PiperOrigin-RevId: 739995908

References

Author

Rifur13

Committer

Google-ML-Automation

Parents

jax a2f22cc1 - [Mosaic GPU] Adding a primitive to load from memrefs *with* a specified layout.