tf.config.set_soft_device_placement

View source on GitHub

Set if soft device placement is enabled.

If enabled, an op will be placed on CPU if any of the following are true

  1. there's no GPU implementation for the OP
  2. no GPU devices are known or registered
  3. need to co-locate with reftype input(s) which are from CPU
Args
enabled Whether to enable soft placement.

© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.3/api_docs/python/tf/config/set_soft_device_placement