[ Web Proxy ]
URL:
Viewing: https://www.tensorflow.org/api_docs/python/tf/debugging/set_log_device_placement [Back]  [Original]
tf.debugging.set_log_device_placement  |  TensorFlow v2.16.1 Skip to main content
TensorFlow [TensorFlow]
Install Learn API Ecosystem Community Why TensorFlow
Search:
GitHub
Overview Python C++ Java More
TensorFlow [TensorFlow]

tf.debugging.set_log_device_placement Stay organized with collections Save and categorize content based on your preferences.

Turns logging for device placement decisions on or off.

View aliases

Compat aliases for migration

See Migration guide for more details.

tf.compat.v1.debugging.set_log_device_placement

tf.debugging.set_log_device_placement(
    enabled
)

Used in the notebooks

Used in the guide

Operations execute on a particular device, producing and consuming tensors on that device. This may change the performance of the operation or require TensorFlow to copy data to or from an accelerator, so knowing where operations execute is useful for debugging performance issues.

For more advanced profiling, use the TensorFlow profiler.

Device placement for operations is typically controlled by a tf.device scope, but there are exceptions, for example operations on a tf.Variable which follow the initial placement of the variable. Turning off soft device placement (with tf.config.set_soft_device_placement) provides more explicit control.

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)

Turning on tf.debugging.set_log_device_placement also logs the placement of ops inside tf.function when the function is called.

enabled Whether to enabled device placement logging.

Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates. Some content is licensed under the numpy license.

Last updated 2024-04-26 UTC.

[[["Easy to understand","easyToUnderstand","thumb-up"],["Solved my problem","solvedMyProblem","thumb-up"],["Other","otherUp","thumb-up"]],[["Missing the information I need","missingTheInformationINeed","thumb-down"],["Too complicated / too many steps","tooComplicatedTooManySteps","thumb-down"],["Out of date","outOfDate","thumb-down"],["Samples / code issue","samplesCodeIssue","thumb-down"],["Other","otherDown","thumb-down"]],["Last updated 2024-04-26 UTC."],[],[]]
Web Proxy Viewer  |  New URL  |  Original Page