tf.experimental.dtensor.preferred_device_type

Возвращает предпочтительный тип устройства для ускорителей.

tf.experimental.dtensor.preferred_device_type() -> str

Возвращаемый тип устройства определяется путем проверки первого присутствующего типа устройства из всех поддерживаемых типов устройств в порядке 'TPU', 'GPU', 'CPU'.