| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | replicate_per_replica_nodes.cc | 28 const std::vector<string>& allowed_devices, in ReplicateNode() argument
35 std::vector<Node*> replicated_nodes(allowed_devices.size()); in ReplicateNode()
36 for (int i = 0; i < allowed_devices.size(); ++i) { in ReplicateNode()
37 const auto& device = allowed_devices.at(i); in ReplicateNode()
179 const std::vector<string>& allowed_devices, in ReplicateNodes() argument
182 TF_RETURN_IF_ERROR(helper->ReplicateNode(n, allowed_devices, graph)); in ReplicateNodes()
243 const std::vector<string>& allowed_devices = in ReplicatePerReplicaNodesInFunctionGraph() local
245 if (allowed_devices.empty()) { in ReplicatePerReplicaNodesInFunctionGraph()
250 if (allowed_devices.size() == 1) { in ReplicatePerReplicaNodesInFunctionGraph()
253 n->set_assigned_device_name(allowed_devices.at(0)); in ReplicatePerReplicaNodesInFunctionGraph()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | client_library.cc | 30 const absl::optional<std::set<int>>& allowed_devices) in LocalClientOptions() argument
34 allowed_devices_(allowed_devices) {} in LocalClientOptions()
64 const absl::optional<std::set<int>>& allowed_devices) { in set_allowed_devices() argument
65 allowed_devices_ = allowed_devices; in set_allowed_devices()
69 const absl::optional<std::set<int>>& LocalClientOptions::allowed_devices() in allowed_devices() function in xla::LocalClientOptions
111 service_options.set_allowed_devices(options.allowed_devices()); in GetOrCreateLocalClient()
|
| D | client_library.h | 51 const absl::optional<std::set<int>>& allowed_devices = absl::nullopt);
69 const absl::optional<std::set<int>>& allowed_devices);
70 const absl::optional<std::set<int>>& allowed_devices() const;
90 const absl::optional<std::set<int>>& allowed_devices = absl::nullopt);
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/ |
| D | tf_saved_model_lift_variables_invalid_session.mlir | 9 …%0 = "tf.VarHandleOp"() {_class = ["loc:@dense/kernel"], allowed_devices = [], container = "", dev…
11 …%2 = "tf.VarHandleOp"() {_class = ["loc:@dense/bias"], allowed_devices = [], container = "", devic…
26 …%0 = "tf.VarHandleOp"() {_class = ["loc:@dense/kernel"], allowed_devices = [], container = "", dev…
28 …%2 = "tf.VarHandleOp"() {_class = ["loc:@dense/bias"], allowed_devices = [], container = "", devic…
|
| D | tf_saved_model_lift_variables.mlir | 9 …%0 = "tf.VarHandleOp"() {_class = ["loc:@dense/kernel"], allowed_devices = [], container = "", dev…
11 …%2 = "tf.VarHandleOp"() {_class = ["loc:@dense/bias"], allowed_devices = [], container = "", devic…
33 …%0 = "tf.VarHandleOp"() {_class = ["loc:@dense/kernel"], allowed_devices = [], container = "", dev…
35 …%2 = "tf.VarHandleOp"() {_class = ["loc:@dense/bias"], allowed_devices = [], container = "", devic…
43 …%0 = "tf.VarHandleOp"() {_class = ["loc:@dense/kernel"], allowed_devices = [], container = "", dev…
45 …%2 = "tf.VarHandleOp"() {_class = ["loc:@dense/bias"], allowed_devices = [], container = "", devic…
|
| D | resource_inlining.mlir | 15 …%0 = "tf.VarHandleOp"() {_class = ["loc:@Variable"], allowed_devices = [], container = "", device …
|
| D | mark_ops_for_outside_compilation.mlir | 355 …"tf.VarHandleOp"() {allowed_devices = [], container = "", device = "", shared_name = ""} : () -> t…
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | backend.cc | 60 const absl::optional<std::set<int>>& allowed_devices) { in set_allowed_devices() argument
61 allowed_devices_ = allowed_devices; in set_allowed_devices()
65 const absl::optional<std::set<int>>& BackendOptions::allowed_devices() const { in allowed_devices() function in xla::BackendOptions
88 PlatformUtil::GetStreamExecutors(platform, options.allowed_devices())); in CreateBackend()
|
| D | platform_util.cc | 167 const absl::optional<std::set<int>>& allowed_devices) { in GetStreamExecutors() argument
194 if (allowed_devices && allowed_devices->count(i) == 0) { in GetStreamExecutors()
|
| D | backend.h | 61 const absl::optional<std::set<int>>& allowed_devices);
62 const absl::optional<std::set<int>>& allowed_devices() const;
|
| D | platform_util.h | 59 const absl::optional<std::set<int>>& allowed_devices = absl::nullopt);
|
| D | service.h | 68 const absl::optional<std::set<int>>& allowed_devices);
69 const absl::optional<std::set<int>>& allowed_devices() const;
|
| D | local_service.cc | 58 .set_allowed_devices(options.allowed_devices()); in NewService()
|
| D | service.cc | 125 const absl::optional<std::set<int>>& allowed_devices) { in set_allowed_devices() argument
126 allowed_devices_ = allowed_devices; in set_allowed_devices()
130 const absl::optional<std::set<int>>& ServiceOptions::allowed_devices() const { in allowed_devices() function in xla::ServiceOptions
150 backend_options.set_allowed_devices(options.allowed_devices()); in NewService()
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | VarHandleOp.pbtxt | 60 name: "allowed_devices"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | VarHandleOp.pbtxt | 60 name: "allowed_devices"
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_VarHandleOp.pbtxt | 29 name: "allowed_devices"
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_gpu_device.cc | 156 options.allowed_devices = gpu_ids; in CreateDevices()
|
| D | xla_device.h | 137 absl::optional<std::set<int>> allowed_devices; member
|
| D | xla_device.cc | 214 allowed_devices_(options.allowed_devices) { in XlaDevice()
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 5277 …argspec: "args=[\'dtype\', \'shape\', \'container\', \'shared_name\', \'allowed_devices\', \'name\…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 5277 …argspec: "args=[\'dtype\', \'shape\', \'container\', \'shared_name\', \'allowed_devices\', \'name\…
|
| /external/llvm-project/llvm/lib/Analysis/models/inliner/ |
| D | saved_model.pbtxt | 181 name: "allowed_devices"
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 58062 name: "allowed_devices"
|