tf.debugging.set_log_device_placement

Включает или отключает протоколирование решений о размещении устройства.

tf.debugging.set_log_device_placement(
    enabled
)

Операции выполняются на конкретном устройстве, создавая и потребляя тензоры на этом устройстве. Это может повлиять на производительность операции или потребовать от TensorFlow копирования данных на или с ускорителя, поэтому знание места выполнения операций полезно для отладки проблем с производительностью.

Для более продвинутого профилирования используйте профилировщик TensorFlow.

Размещение устройства для операций обычно контролируется tf.device областью, но есть исключения, например, операции на tf.Variable, которые следуют начальному размещению переменной. Отключение гибкого размещения устройства (с помощью tf.config.set_soft_device_placement) обеспечивает более явный контроль.

tf.debugging.set_log_device_placement(True)
tf.ones([])
# [...] op Fill in device /job:localhost/replica:0/task:0/device:GPU:0
with tf.device("CPU"):
 tf.ones([])
# [...] op Fill in device /job:localhost/replica:0/task:0/device:CPU:0
tf.debugging.set_log_device_placement(False)

Включение tf.debugging.set_log_device_placement также записывает размещение операций внутри tf.function при вызове функции.