xla
46d47d35 - [PJRT] Detect number of local TPU chips (#4720)

Commit

2 years ago

[PJRT] Detect number of local TPU chips (#4720) * [PJRT] Detect number of local TPU chips * Update docstring

References

Author

will-cromar

Parents