jax

7d6fd6da - [Mosaic:TPU] Fold tpu.erase_layout into memref.load and memref.store

[Mosaic:TPU] Fold tpu.erase_layout into memref.load and memref.store Unlike the vector dialect versions of these ops, the verifier has no issues with tiled layouts. PiperOrigin-RevId: 866669577