Home
last modified time | relevance | path

Searched refs:set_soft_device_placement (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/python/eager/
Ddevice_placement_test.py40 config.set_soft_device_placement(enabled=True)
149 config.set_soft_device_placement(enabled=False)
221 config.set_soft_device_placement(enabled=True)
Dops_test.py301 config.set_soft_device_placement(True)
Dcontext.py2346 def set_soft_device_placement(enabled): function
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.__internal__.eager_context.pbtxt20 name: "set_soft_device_placement"
Dtensorflow.config.pbtxt76 name: "set_soft_device_placement"
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.config.pbtxt76 name: "set_soft_device_placement"
/external/tensorflow/tensorflow/tools/docs/
Dtf_doctest.py154 tf.config.set_soft_device_placement(True)
/external/tensorflow/tensorflow/python/framework/
Dconfig_test.py165 config.set_soft_device_placement(True)
175 config.set_soft_device_placement(False)
436 config.set_soft_device_placement(False)
530 config.set_soft_device_placement(False)
Dconfig.py263 def set_soft_device_placement(enabled): function
/external/tensorflow/tensorflow/python/distribute/
Dtpu_strategy_model_parallelism_test.py371 config.set_soft_device_placement(True)
393 config.set_soft_device_placement(original_device_placement)
Dtpu_strategy_test.py119 config.set_soft_device_placement(True)
397 config.set_soft_device_placement(False)
424 config.set_soft_device_placement(True)
/external/tensorflow/tensorflow/python/tpu/
Dtpu_outside_compilation_test.py130 config.set_soft_device_placement(False)
522 config.set_soft_device_placement(True)
/external/tensorflow/tensorflow/compiler/tests/
Dstateless_random_ops_test.py350 config.set_soft_device_placement(False)
Dstateful_random_ops_test.py357 config.set_soft_device_placement(False)
/external/tensorflow/tensorflow/python/kernel_tests/random/
Dstateless_random_ops_test.py551 config.set_soft_device_placement(False)
Dstateful_random_ops_test.py718 config.set_soft_device_placement(False)