jax
a16fbffc - [Mosaic][TPU] Add a compatibility mode to Mosaic's canonicalization pass, skipping over elementwise and matmul op insertions and/or type compat casts.

Commit

1 year ago

[Mosaic][TPU] Add a compatibility mode to Mosaic's canonicalization pass, skipping over elementwise and matmul op insertions and/or type compat casts. PiperOrigin-RevId: 714132282

References

#213 - CI: 01/27/25 upstream sync

Author

Google-ML-Automation

Committer

Google-ML-Automation

Parents

743872df

jax a16fbffc - [Mosaic][TPU] Add a compatibility mode to Mosaic's canonicalization pass, skipping over elementwise and matmul op insertions and/or type compat casts.

jax
a16fbffc - [Mosaic][TPU] Add a compatibility mode to Mosaic's canonicalization pass, skipping over elementwise and matmul op insertions and/or type compat casts.