TF 2.0 is out! Get hands-on practice at TF World, Oct 28-31. Use code TF20 for 20% off select passes. Register now

tf.config.set_soft_device_placement

TensorFlow 1 version View source on GitHub

Set if soft device placement is enabled.

Aliases:

tf.config.set_soft_device_placement(enabled)

Used in the guide:

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.