View source on GitHub |
Return status of soft device placement flag.
tf.config.get_soft_device_placement()
If enabled, an op will be placed on CPU if any of the following are true
If disabled, the placement is strict and CPU fallback is not allowed. An error is raised when an Op cannot be placed onto its intended device.
Returns | |
---|---|
A boolean indicating if soft placement is enabled. |
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.9/api_docs/python/tf/config/get_soft_device_placement