TensorFlow 2.0 Beta is available Learn more

tf.config.get_soft_device_placement

TensorFlow 2.0 version View source on GitHub

Get if soft device placement is enabled.

Aliases:

  • tf.compat.v1.config.get_soft_device_placement
  • tf.compat.v2.config.get_soft_device_placement
  • tf.config.get_soft_device_placement
tf.config.get_soft_device_placement()

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

Returns:

If soft placement is enabled.