jax
6a6ceee6 - [Mosaic:TPU] 16-bit tpu.dynamic_gather

Commit

1 year ago

[Mosaic:TPU] 16-bit tpu.dynamic_gather PiperOrigin-RevId: 776046842

Author

tlongeri

tlongeri

Committer

Google-ML-Automation

Google-ML-Automation

Parents

Loading