jax
b9abdcb8 - [Mosaic] Upcast lhs to bf16 for f8 @ bf16/f32 on TPU 7x.

Commit

204 days ago

[Mosaic] Upcast lhs to bf16 for f8 @ bf16/f32 on TPU 7x. PiperOrigin-RevId: 829512043

References

Author

WindQAQ

Committer

Google-ML-Automation

Parents