Searched refs:assigned_device (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | placer.cc | 120 Status AssignAndLog(int assigned_device, Node* node, in AssignAndLog() argument
123 node->set_assigned_device_name_index(assigned_device); in AssignAndLog()
223 int assigned_device = -1; in Run() local
237 assigned_device = input->assigned_device_name_index(); in Run()
242 if (assigned_device == -1) { in Run()
243 assigned_device = graph_->InternDeviceName((*devices)[0]->name()); in Run()
246 TF_RETURN_IF_ERROR(AssignAndLog(assigned_device, node, &colocation_graph, in Run()
262 int assigned_device = -1; in Run() local
277 assigned_device = output_device_name; in Run()
282 if (assigned_device == -1) { in Run()
[all …]
|
| D | colocation_graph.cc | 1399 const Device* assigned_device = in InitializeMemberWithAssignedDevice() local
1401 if (assigned_device == nullptr) { in InitializeMemberWithAssignedDevice()
1416 if (IsSupportedDeviceType(assigned_device->attributes(), d.first)) { in InitializeMemberWithAssignedDevice()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | node_matchers.cc | 122 if (assigned_device && node->assigned_device_name() != *assigned_device) { in MatchAndExplain()
124 *listener << "\nexpected assigned_device " << *assigned_device in MatchAndExplain()
238 if (assigned_device) { in DescribeTo()
239 predicates.push_back(absl::StrCat("assigned device: ", *assigned_device)); in DescribeTo()
332 absl::optional<string> assigned_device; member
410 if (prop.assigned_device()) { in NodeWith()
411 DCHECK(!matcher->assigned_device); in NodeWith()
412 matcher->assigned_device = prop.assigned_device(); in NodeWith()
449 impl::NodeMatcherProperties AssignedDevice(string assigned_device) { in AssignedDevice() argument
451 props.set_assigned_device(std::move(assigned_device)); in AssignedDevice()
|
| D | node_matchers.h | 91 const absl::optional<string>& assigned_device() const { in assigned_device() function
115 void set_assigned_device(string assigned_device) { in set_assigned_device() argument
117 assigned_device_ = std::move(assigned_device); in set_assigned_device()
188 impl::NodeMatcherProperties AssignedDevice(string assigned_device);
|
| /external/tensorflow/tensorflow/cc/framework/ |
| D | scope.h | 144 Scope WithAssignedDevice(const string& assigned_device) const;
|
| D | scope.cc | 235 const string& assigned_device) in Impl() argument
247 assigned_device_(assigned_device), in Impl()
453 Scope Scope::WithAssignedDevice(const string& assigned_device) const { in WithAssignedDevice()
454 return Scope(new Impl(*this, Impl::Tags::AssignedDevice(), assigned_device)); in WithAssignedDevice()
|
| D | scope_internal.h | 81 Impl(const Scope& other, Tags::AssignedDevice, const string& assigned_device);
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_pass.cc | 1084 const string& assigned_device = unpadded_input->assigned_device_name(); in SetPaddingNodesDevices() local
1085 if (!requested_device.empty() || !assigned_device.empty()) { in SetPaddingNodesDevices()
1094 out->set_assigned_device_name(assigned_device); in SetPaddingNodesDevices()
1102 out->set_assigned_device_name(assigned_device); in SetPaddingNodesDevices()
|