View source on GitHub |
Returns the preferred device type for the accelerators.
tf.experimental.dtensor.preferred_device_type() -> str
The returned device type is determined by checking the first present device type from all supported device types in the order of 'TPU', 'GPU', 'CPU'.