tf.experimental.dtensor.preferred_device_type

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

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

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