Searched refs:assigned_device (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | placer.cc | 67 Status AssignAndLog(int assigned_device, Node* node, in AssignAndLog() argument
70 node->set_assigned_device_name_index(assigned_device); in AssignAndLog()
161 int assigned_device = -1; in Run() local
175 assigned_device = input->assigned_device_name_index(); in Run()
180 if (assigned_device == -1) { in Run()
181 assigned_device = graph_->InternDeviceName((*devices)[0]->name()); in Run()
184 TF_RETURN_IF_ERROR(AssignAndLog(assigned_device, node, &colocation_graph, in Run()
200 int assigned_device = -1; in Run() local
215 assigned_device = output_device_name; in Run()
220 if (assigned_device == -1) { in Run()
[all …]
|
| D | colocation_graph.cc | 921 const Device* assigned_device = in InitializeMemberWithAssignedDevice() local
923 if (assigned_device == nullptr) { in InitializeMemberWithAssignedDevice()
929 if (DeviceType(assigned_device->attributes().device_type()) == d.first) { in InitializeMemberWithAssignedDevice()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | node_matchers.cc | 118 if (assigned_device && node->assigned_device_name() != *assigned_device) { in MatchAndExplain()
120 *listener << "\nexpected assigned_device " << *assigned_device in MatchAndExplain()
234 if (assigned_device) { in DescribeTo()
235 predicates.push_back(absl::StrCat("assigned device: ", *assigned_device)); in DescribeTo()
328 absl::optional<string> assigned_device; member
406 if (prop.assigned_device()) { in NodeWith()
407 DCHECK(!matcher->assigned_device); in NodeWith()
408 matcher->assigned_device = prop.assigned_device(); in NodeWith()
445 impl::NodeMatcherProperties AssignedDevice(string assigned_device) { in AssignedDevice() argument
447 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.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.h | 144 Scope WithAssignedDevice(const string& assigned_device) const;
|
| D | scope_internal.h | 81 Impl(const Scope& other, Tags::AssignedDevice, const string& assigned_device);
|