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
- there's no GPU implementation for the OP
- no GPU devices are known or registered
- need to co-locate with reftype input(s) which are from CPU
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. |