tf.config.set_soft_device_placement

Enable or disable soft device placement.

tf.config.set_soft_device_placement(
    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

enabled A boolean indicating whether to enable soft placement.