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 при вызове функции.