tf.debugging.set_log_device_placement

TensorFlow 2 version View source on GitHub

Set if device placements should be logged.

enabled Whether to enabled device placement logging.