xla
ef95cb41 - [PJRT] Add `tpu.num_available_devices` to get number of XLA TPU devices (#4927)

Commit

2 years ago

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

References

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

Author

will-cromar

Parents

045aa264

xla ef95cb41 - [PJRT] Add `tpu.num_available_devices` to get number of XLA TPU devices (#4927)

xla
ef95cb41 - [PJRT] Add `tpu.num_available_devices` to get number of XLA TPU devices (#4927)