tf.config.set_soft_device_placement

View source on GitHub

Enable or disable soft device placement.

View aliases

Compat aliases for migration

See Migration guide for more details.

tf.compat.v1.config.set_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

Note: by default soft device placement is enabled when running in eager mode (for convenience) and disabled in graph mode (for performance).

Args
enabled	A boolean indicating whether to enable soft placement.