jax
91651f80 - [Mosaic] Use BF16 ops for math::PowF on TPUv6+.

Commit

229 days ago

[Mosaic] Use BF16 ops for math::PowF on TPUv6+. PiperOrigin-RevId: 772444344

References

#29214 - [Mosaic] Use BF16 ops for math::PowF on TPUv6+.

Author

WindQAQ

WindQAQ

Committer

Google-ML-Automation

Google-ML-Automation

Parents

Loading