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'.
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/experimental/dtensor/preferred_device_type