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