jax
8954e71d - [Mosaic TPU] Improve support for int16->int32 casts in TPUv4

Commit

349 days ago

[Mosaic TPU] Improve support for int16->int32 casts in TPUv4 PiperOrigin-RevId: 716250236

References

Author

apaszke

Committer

Google-ML-Automation

Parents