Home
last modified time | relevance | path

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

123

/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/compiler/mlir/tensorflow/tests/
Dmark_ops_for_outside_compilation.mlir27 …}) {allow_soft_placement = false, num_cores_per_replica = 1, topology = "", device_assignment = …
45 …}) {allow_soft_placement = true, num_cores_per_replica = 1, topology = "", device_assignment = […
59 …}) {allow_soft_placement = true, num_cores_per_replica = 1, topology = "", device_assignment = […
77 …}) {allow_soft_placement = true, num_cores_per_replica = 1, topology = "", device_assignment = […
91 …}) {allow_soft_placement = true, num_cores_per_replica = 1, topology = "", device_assignment = […
102 …}) {allow_soft_placement = true, num_cores_per_replica = 1, topology = "", device_assignment = […
116 …}) {allow_soft_placement = true, num_cores_per_replica = 1, topology = "", device_assignment = […
134 …}) {allow_soft_placement = true, num_cores_per_replica = 1, topology = "", device_assignment = […
151 …}) {allow_soft_placement = true, num_cores_per_replica = 1, topology = "", device_assignment = […
169 …}) {allow_soft_placement = true, num_cores_per_replica = 1, topology = "", device_assignment = […
[all …]
Dgraph_pruning_preserve_ops.mlir12 …%2 = tf_executor.island(%1) wraps "tf.TPUReplicateMetadata"() {allow_soft_placement = false, compu…
/external/tensorflow/tensorflow/core/common_runtime/
Dcolocation_graph.h81 Status MergeDeviceNames(const Member& other, bool allow_soft_placement);
97 bool allow_soft_placement);
220 const Device* default_local_device, bool allow_soft_placement,
Dplacer_test.cc261 bool allow_soft_placement, bool log_device_placement) { in Place() argument
263 allow_soft_placement, log_device_placement); in Place()
268 bool allow_soft_placement, in CallOptPassesAndPlace() argument
300 allow_soft_placement, log_device_placement); in CallOptPassesAndPlace()
308 Status Place(Graph* graph, bool allow_soft_placement, in Place() argument
310 return Place(graph, &devices_, nullptr, allow_soft_placement, in Place()
318 Status CallOptPassesAndPlace(Graph* graph, bool allow_soft_placement, in CallOptPassesAndPlace() argument
320 return CallOptPassesAndPlace(graph, &devices_, allow_soft_placement, in CallOptPassesAndPlace()
1038 auto handle_test = [this](bool allow_soft_placement, bool set_assigned) { in TEST_F() argument
1065 Status s = Place(&g, allow_soft_placement, true); in TEST_F()
[all …]
Dinspecting_placer.cc117 bool allow_soft_placement, in InspectingPlacer() argument
123 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,
Dplacer.cc137 bool allow_soft_placement, bool log_device_placement) in Placer() argument
143 allow_soft_placement_(allow_soft_placement), in Placer()
Dcolocation_graph.cc369 bool allow_soft_placement) { in MergeDeviceNames() argument
385 allow_soft_placement)); in MergeDeviceNames()
534 bool allow_soft_placement) { in LimitToPossibleDevices() argument
537 allow_soft_placement)); in LimitToPossibleDevices()
608 bool allow_soft_placement, in ColocationGraph() argument
614 allow_soft_placement, log_device_placement), in ColocationGraph()
621 allow_soft_placement_(allow_soft_placement), in ColocationGraph()
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
Dcopy_to_device_test.py309 config=config_pb2.ConfigProto(allow_soft_placement=False)):
333 config=config_pb2.ConfigProto(allow_soft_placement=False)):
353 config=config_pb2.ConfigProto(allow_soft_placement=False)):
373 config=config_pb2.ConfigProto(allow_soft_placement=False)):
393 config=config_pb2.ConfigProto(allow_soft_placement=False)):
415 config=config_pb2.ConfigProto(allow_soft_placement=False)):
494 config=config_pb2.ConfigProto(allow_soft_placement=False)):
518 config=config_pb2.ConfigProto(allow_soft_placement=False)):
543 config=config_pb2.ConfigProto(allow_soft_placement=False)):
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DTPUReplicateMetadata.pbtxt198 name: "allow_soft_placement"
273 name: "allow_soft_placement"
/external/tensorflow/tensorflow/python/tpu/
Dtpu_strategy_util.py119 session_config = config_pb2.ConfigProto(allow_soft_placement=True)
206 session_config = config_pb2.ConfigProto(allow_soft_placement=True)
/external/tensorflow/tensorflow/core/util/
Ddevice_name_utils.cc349 bool allow_soft_placement, bool override_conflicts) { in MergeDevNamesImpl() argument
389 if (!allow_soft_placement) { in MergeDevNamesImpl()
409 if (!allow_soft_placement) { in MergeDevNamesImpl()
434 bool allow_soft_placement) { in MergeDevNames() argument
435 return MergeDevNamesImpl(target, other, allow_soft_placement, in MergeDevNames()
Ddevice_name_utils_test.cc401 bool allow_soft_placement) { in MergeDevNamesHelperImpl() argument
404 allow_soft_placement)); in MergeDevNamesHelperImpl()
407 allow_soft_placement)); in MergeDevNamesHelperImpl()
Ddevice_name_utils.h151 bool allow_soft_placement);
/external/tensorflow/tensorflow/python/kernel_tests/
Drnn_test.py405 config.allow_soft_placement = True
457 config.allow_soft_placement = True
520 config.allow_soft_placement = True
588 config.allow_soft_placement = True
653 config.allow_soft_placement = True
688 config.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/compiler/mlir/tensorflow/tests/tpu_bridge_v1/
Dend_to_end.mlir12 …nd wraps "tf.TPUReplicateMetadata"() {_tpu_replicate = "cluster", allow_soft_placement = false, co…
/external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/
Dtrt_engine_op.cc108 bool allow_soft_placement = false, size_t num_inputs = 0,
278 bool allow_soft_placement, size_t num_inputs, size_t num_outputs) { in ConstructFunctionHandle() argument
286 if (allow_soft_placement) { in ConstructFunctionHandle()
/external/tensorflow/tensorflow/python/ops/
Dbatch_ops_test.py333 config = config_pb2.ConfigProto(allow_soft_placement=True)
339 config.allow_soft_placement = False
/external/tensorflow/tensorflow/python/eager/
Dcontext.py956 config.allow_soft_placement = self._soft_device_placement
958 config.allow_soft_placement = self.executing_eagerly()
1108 config.allow_soft_placement = True
1646 return self.config.allow_soft_placement
/external/tensorflow/tensorflow/python/framework/
Dtest_util.py3229 allow_soft_placement = not force_gpu
3232 config.allow_soft_placement = allow_soft_placement
3233 elif not allow_soft_placement and config.allow_soft_placement:
3236 config.allow_soft_placement = False

123