jax
fbf4610f - [Mosaic] Add `sign_extended` attribute to `tpu::UnpackSubelementsOp`.

Commit

120 days ago

[Mosaic] Add `sign_extended` attribute to `tpu::UnpackSubelementsOp`. This is useful if we don't care extended bits. PiperOrigin-RevId: 818148251

References

Author

WindQAQ

Committer

Google-ML-Automation

Parents