jax
29cb6cd1 - [Mosaic TPU] Add MemRead and MemStore effects to load and store ops.

Commit

310 days ago

[Mosaic TPU] Add MemRead and MemStore effects to load and store ops. So duplicated load/store ops can be removed. PiperOrigin-RevId: 745209849

References

Author

bythew3i

Committer

Google-ML-Automation

Parents