tf.experimental.dtensor.device_name

Возвращает имя устройства DTensor.

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

Эта функция может быть использована следующим образом:

import tensorflow as tf

with tf.device(dtensor.device_name()):
  # ...