jax
75d6129e - [mosaic] Clarified that tpu.device_id return a logical ID

Commit

1 year ago

[mosaic] Clarified that tpu.device_id return a logical ID PiperOrigin-RevId: 781022013

Author

superbobry

Committer

Google-ML-Automation

Parents