Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/python/grappler/
Dcluster.py33 allow_soft_placement=True, argument
53 self._tf_cluster = tf_cluster.TF_NewCluster(allow_soft_placement,
115 def Provision(allow_soft_placement=True, argument
119 cluster = Cluster(allow_soft_placement, disable_detailed_stats,
Dcluster_wrapper.cc78 [](bool allow_soft_placement, in PYBIND11_MODULE()
90 cluster->AllowSoftPlacement(allow_soft_placement); in PYBIND11_MODULE()
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
Dcopy_to_device_test.py271 config=config_pb2.ConfigProto(allow_soft_placement=False)):
292 config=config_pb2.ConfigProto(allow_soft_placement=False)):
327 config=config_pb2.ConfigProto(allow_soft_placement=False)):
351 config=config_pb2.ConfigProto(allow_soft_placement=False)):
371 config=config_pb2.ConfigProto(allow_soft_placement=False)):
391 config=config_pb2.ConfigProto(allow_soft_placement=False)):
411 config=config_pb2.ConfigProto(allow_soft_placement=False)):
433 config=config_pb2.ConfigProto(allow_soft_placement=False)):
512 config=config_pb2.ConfigProto(allow_soft_placement=False)):
536 config=config_pb2.ConfigProto(allow_soft_placement=False)):
[all …]
Dprefetch_to_device_test.py155 config=config_pb2.ConfigProto(allow_soft_placement=False)):
203 config=config_pb2.ConfigProto(allow_soft_placement=False)):
/external/tensorflow/tensorflow/core/common_runtime/
Dcolocation_graph.h77 Status MergeDeviceNames(const Member& other, bool allow_soft_placement);
93 bool allow_soft_placement);
216 const Device* default_local_device, bool allow_soft_placement,
Dplacer_test.cc258 bool allow_soft_placement, bool log_device_placement) { in Place() argument
260 allow_soft_placement, log_device_placement); in Place()
265 bool allow_soft_placement, in CallOptPassesAndPlace() argument
297 allow_soft_placement, log_device_placement); in CallOptPassesAndPlace()
305 Status Place(Graph* graph, bool allow_soft_placement, in Place() argument
307 return Place(graph, &devices_, nullptr, allow_soft_placement, in Place()
315 Status CallOptPassesAndPlace(Graph* graph, bool allow_soft_placement, in CallOptPassesAndPlace() argument
317 return CallOptPassesAndPlace(graph, &devices_, allow_soft_placement, in CallOptPassesAndPlace()
1035 auto handle_test = [this](bool allow_soft_placement, bool set_assigned) { in TEST_F() argument
1062 Status s = Place(&g, allow_soft_placement, true); in TEST_F()
[all …]
Dinspecting_placer.cc116 bool allow_soft_placement, in InspectingPlacer() argument
122 allow_soft_placement_(allow_soft_placement), in InspectingPlacer()
Dinspecting_placer.h75 bool allow_soft_placement, bool log_device_placement);
Dplacer.h73 const Device* default_local_device, bool allow_soft_placement,
Dcolocation_graph.cc342 bool allow_soft_placement) { in MergeDeviceNames() argument
358 allow_soft_placement)); in MergeDeviceNames()
504 bool allow_soft_placement) { in LimitToPossibleDevices() argument
507 allow_soft_placement)); in LimitToPossibleDevices()
578 bool allow_soft_placement, in ColocationGraph() argument
584 allow_soft_placement, log_device_placement), in ColocationGraph()
591 allow_soft_placement_(allow_soft_placement), in ColocationGraph()
Dplacer.cc137 bool allow_soft_placement, bool log_device_placement) in Placer() argument
143 allow_soft_placement_(allow_soft_placement), in Placer()
/external/tensorflow/tensorflow/core/util/
Ddevice_name_utils.cc344 bool allow_soft_placement, bool override_conflicts) { in MergeDevNamesImpl() argument
384 if (!allow_soft_placement) { in MergeDevNamesImpl()
404 if (!allow_soft_placement) { in MergeDevNamesImpl()
429 bool allow_soft_placement) { in MergeDevNames() argument
430 return MergeDevNamesImpl(target, other, allow_soft_placement, in MergeDevNames()
Ddevice_name_utils_test.cc397 bool allow_soft_placement) { in MergeDevNamesHelperImpl() argument
400 allow_soft_placement)); in MergeDevNamesHelperImpl()
403 allow_soft_placement)); in MergeDevNamesHelperImpl()
Ddevice_name_utils.h142 bool allow_soft_placement);
/external/tensorflow/tensorflow/python/tpu/
Dtpu_strategy_util.py112 session_config = config_pb2.ConfigProto(allow_soft_placement=True)
193 session_config = config_pb2.ConfigProto(allow_soft_placement=True)
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
DTPUReplicateMetadata.pbtxt198 name: "allow_soft_placement"
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.-config-proto.pbtxt57 name: "allow_soft_placement"
/external/tensorflow/tensorflow/python/kernel_tests/
Drnn_test.py762 config.allow_soft_placement = True
814 config.allow_soft_placement = True
877 config.allow_soft_placement = True
945 config.allow_soft_placement = True
1010 config.allow_soft_placement = True
1045 config.allow_soft_placement = True
Dsparse_xent_op_test.py300 config.allow_soft_placement = True
/external/tensorflow/tensorflow/python/eager/
Dcontext.py861 config.allow_soft_placement = self._soft_device_placement
863 config.allow_soft_placement = self.executing_eagerly()
999 config.allow_soft_placement = True
1416 return self.config.allow_soft_placement
/external/tensorflow/tensorflow/python/framework/
Dtest_util.py2976 allow_soft_placement = not force_gpu
2979 config.allow_soft_placement = allow_soft_placement
2980 elif not allow_soft_placement and config.allow_soft_placement:
2983 config.allow_soft_placement = False
/external/tensorflow/tensorflow/compiler/tests/
Dxla_test.py290 config = config_pb2.ConfigProto(allow_soft_placement=True)
/external/tensorflow/tensorflow/python/ops/
Dinit_ops_test.py195 allow_soft_placement=False, log_device_placement=True)
/external/tensorflow/tensorflow/python/distribute/
Dmulti_worker_test_base.py316 config = config_pb2.ConfigProto(allow_soft_placement=True)
Ddistribute_coordinator.py770 allow_soft_placement=True)

12