xla
[PJRT] Add `tpu.num_available_devices` to get number of XLA TPU devices
#4927
Merged

[PJRT] Add `tpu.num_available_devices` to get number of XLA TPU devices #4927

JackCaoG merged 2 commits into master from wcromar/num-available-devices
will-cromar
will-cromar [PJRT] Add `tpu.num_available_devices` to get number of XLA TPU devices
b52b103e
will-cromar docstrings
46ee89a1
will-cromar will-cromar requested a review from Liyang90 Liyang90 2 years ago
will-cromar will-cromar requested a review from JackCaoG JackCaoG 2 years ago
Liyang90
Liyang90 commented on 2023-04-21
JackCaoG
JackCaoG approved these changes on 2023-04-21
JackCaoG JackCaoG merged ef95cb41 into master 2 years ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone