jax
094b66fa - [Mosaic GPU] Enable transpose tests in mosaic_gpu.

Commit

321 days ago

[Mosaic GPU] Enable transpose tests in mosaic_gpu. Not sure why these were disabled, but they work now. PiperOrigin-RevId: 770792116

References

Author

dimitar-asenov

Committer

Google-ML-Automation

Parents