Home
last modified time | relevance | path

Searched refs:allow_soft_placement (Results 1 – 25 of 44) sorted by relevance

12

/external/tensorflow/tensorflow/python/grappler/
Dcluster.py34 allow_soft_placement=True, argument
55 allow_soft_placement, disable_detailed_stats, status)
123 def Provision(allow_soft_placement=True, argument
127 cluster = Cluster(allow_soft_placement, disable_detailed_stats,
Dcluster.i133 static GCluster TF_NewCluster(bool allow_soft_placement, in TF_NewCluster() argument
142 cluster_->AllowSoftPlacement(allow_soft_placement); in TF_NewCluster()
437 bool allow_soft_placement, bool disable_detailed_stats, TF_Status* out_status);
/external/tensorflow/tensorflow/core/common_runtime/
Dcolocation_graph.h67 Status MergeDeviceNames(const Member& other, bool allow_soft_placement);
74 Status AssignDevice(const Node& node, bool allow_soft_placement);
171 const Device* default_device, bool allow_soft_placement,
Dplacer_test.cc242 Status Place(Graph* graph, DeviceSet* devices, bool allow_soft_placement, in Place() argument
244 Placer placer(graph, devices, nullptr, allow_soft_placement, in Place()
253 Status Place(Graph* graph, bool allow_soft_placement, in Place() argument
255 return Place(graph, &devices_, allow_soft_placement, log_device_placement); in Place()
905 auto handle_test = [this](bool allow_soft_placement, bool set_assigned) { in TEST_F() argument
932 Status s = Place(&g, allow_soft_placement, true); in TEST_F()
1098 bool allow_soft_placement = GetParam(); in TEST_P() local
1099 Status s = Place(&g, allow_soft_placement, true); in TEST_P()
1100 if (allow_soft_placement) { in TEST_P()
1173 bool allow_soft_placement = GetParam(); in TEST_P() local
[all …]
Dcolocation_graph.cc306 bool allow_soft_placement) { in MergeDeviceNames() argument
317 allow_soft_placement)); in MergeDeviceNames()
406 Status Member::AssignDevice(const Node& node, bool allow_soft_placement) { in AssignDevice() argument
414 allow_soft_placement); in AssignDevice()
424 allow_soft_placement); in AssignDevice()
455 bool allow_soft_placement, in ColocationGraph() argument
461 allow_soft_placement_(allow_soft_placement), in ColocationGraph()
Dplacer.cc82 const Device* default_device, bool allow_soft_placement, in Placer() argument
87 allow_soft_placement_(allow_soft_placement), in Placer()
Dplacer.h72 bool allow_soft_placement, bool log_device_placement);
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
Dcopy_to_device_test.py268 config=config_pb2.ConfigProto(allow_soft_placement=False)):
289 config=config_pb2.ConfigProto(allow_soft_placement=False)):
324 config=config_pb2.ConfigProto(allow_soft_placement=False)):
348 config=config_pb2.ConfigProto(allow_soft_placement=False)):
368 config=config_pb2.ConfigProto(allow_soft_placement=False)):
388 config=config_pb2.ConfigProto(allow_soft_placement=False)):
408 config=config_pb2.ConfigProto(allow_soft_placement=False)):
431 config=config_pb2.ConfigProto(allow_soft_placement=False)):
510 config=config_pb2.ConfigProto(allow_soft_placement=False)):
534 config=config_pb2.ConfigProto(allow_soft_placement=False)):
[all …]
Dprefetch_to_device_test.py154 config=config_pb2.ConfigProto(allow_soft_placement=False)):
203 config=config_pb2.ConfigProto(allow_soft_placement=False)):
/external/tensorflow/tensorflow/python/tpu/
Dtpu_config_test.py42 session_config = config_pb2.ConfigProto(allow_soft_placement=True)
73 session_config = config_pb2.ConfigProto(allow_soft_placement=True)
Dtpu_strategy_util.py97 session_config = config_pb2.ConfigProto(allow_soft_placement=True)
Dtpu_config.py244 allow_soft_placement=True, isolate_session_state=True)
/external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/
Drun_config_test.py299 session_config=config_pb2.ConfigProto(allow_soft_placement=True),
306 self.assertEqual(config_pb2.ConfigProto(allow_soft_placement=True),
316 session_config=config_pb2.ConfigProto(allow_soft_placement=False),
/external/tensorflow/tensorflow/core/util/
Ddevice_name_utils_test.cc383 bool allow_soft_placement) { in MergeDevNamesHelperImpl() argument
386 allow_soft_placement)); in MergeDevNamesHelperImpl()
389 allow_soft_placement)); in MergeDevNamesHelperImpl()
Ddevice_name_utils.cc333 bool allow_soft_placement) { in MergeDevNames() argument
372 if (!allow_soft_placement) { in MergeDevNames()
390 if (!allow_soft_placement) { in MergeDevNames()
Ddevice_name_utils.h137 bool allow_soft_placement);
/external/tensorflow/tensorflow/contrib/boosted_trees/estimator_batch/
Dmodel.py94 session_config.allow_soft_placement = True
96 session_config = config_pb2.ConfigProto(allow_soft_placement=True)
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.-config-proto.pbtxt57 name: "allow_soft_placement"
/external/tensorflow/tensorflow/python/eager/
Dcontext.py267 allow_soft_placement=True,
783 return self._config.allow_soft_placement
787 self._config.allow_soft_placement = enabled
/external/tensorflow/tensorflow/python/kernel_tests/
Drnn_test.py761 config.allow_soft_placement = True
813 config.allow_soft_placement = True
876 config.allow_soft_placement = True
944 config.allow_soft_placement = True
1009 config.allow_soft_placement = True
1044 config.allow_soft_placement = True
/external/tensorflow/tensorflow/python/framework/
Dtest_util.py2574 allow_soft_placement = not force_gpu
2577 config.allow_soft_placement = allow_soft_placement
2579 elif not allow_soft_placement and config.allow_soft_placement:
2583 config.allow_soft_placement = False
/external/tensorflow/tensorflow/compiler/tests/
Dxla_test.py271 config = config_pb2.ConfigProto(allow_soft_placement=True)
/external/tensorflow/tensorflow/python/ops/
Dinit_ops_test.py192 allow_soft_placement=False, log_device_placement=True)
/external/tensorflow/tensorflow/contrib/eager/python/examples/revnet/
Dmain_estimator_tpu.py201 allow_soft_placement=True, log_device_placement=True),
/external/tensorflow/tensorflow/contrib/distribute/python/
Dmulti_worker_test_base.py271 config = config_pb2.ConfigProto(allow_soft_placement=True)

12