Home
last modified time | relevance | path

Searched refs:assigned_device_name (Results 1 – 25 of 38) sorted by relevance

12

/external/tensorflow/tensorflow/core/graph/
Dgraph_partition.cc133 if (src->assigned_device_name() == dst->assigned_device_name()) { in NeedSameDeviceSendRecv()
186 builder->Attr("send_device", edge->src()->assigned_device_name()); in SetSendRecvAttrs()
189 opts.get_incarnation(edge->src()->assigned_device_name()))); in SetSendRecvAttrs()
190 builder->Attr("recv_device", edge->dst()->assigned_device_name()); in SetSendRecvAttrs()
217 cast_builder.Device(src->assigned_device_name()).Input(send_from); in AddSend()
243 send_builder.Device(src->assigned_device_name()).Input(send_from); in AddSend()
282 << " on " << src->assigned_device_name() << " in " in AddRecv()
285 << " on " << dst->assigned_device_name() << " in " in AddRecv()
292 << " on " << src->assigned_device_name() << " for " << dst->name() in AddRecv()
294 << " on " << dst->assigned_device_name(); in AddRecv()
[all …]
Dmkl_tfconversion_pass.cc189 conversion_node->set_assigned_device_name(src->assigned_device_name()); in InsertConversionNodeOnEdge()
275 conversion_node->set_assigned_device_name(n->assigned_device_name()); in InsertInputConversionNode()
Dcostmodel.cc370 if (persistent_alloc_ids_by_devices_.find(node->assigned_device_name()) == in IsPersistentTensor()
374 return persistent_alloc_ids_by_devices_.at(node->assigned_device_name()) in IsPersistentTensor()
485 cnode->set_device(n->assigned_device_name()); in AddToCostGraphDef()
Dgraph.cc129 strings::StrAppend(&ret, "{", assigned_device_name(), "}"); in DebugString()
453 copy->set_assigned_device_name(node->assigned_device_name()); in CopyNode()
680 if (!node->assigned_device_name().empty()) { in ToGraphDefSubRange()
681 node_def->set_device(node->assigned_device_name()); in ToGraphDefSubRange()
Dmkl_layout_pass.cc1050 if (!n->assigned_device_name().empty() && in CanOpRunOnCPUDevice()
1051 !absl::StrContains(n->assigned_device_name(), kCPUDeviceSubStr)) { in CanOpRunOnCPUDevice()
2014 (*out)->set_assigned_device_name(orig_node->assigned_device_name()); in GetDummyMklTensorNode()
2948 pred->assigned_device_name() != succ->assigned_device_name() || in MergeConv2DWithBiasAdd()
3065 new_node->set_assigned_device_name(pred->assigned_device_name()); in MergeConv2DWithBiasAdd()
3108 pred->assigned_device_name() != succ->assigned_device_name() || in MergePadWithConv2D()
3241 new_node->set_assigned_device_name(pred->assigned_device_name()); in MergePadWithConv2D()
3273 badd->assigned_device_name() != fltr->assigned_device_name() || in MergeConv2DBackpropFilterWithBiasAddGrad()
3373 new_node->set_assigned_device_name(badd->assigned_device_name()); in MergeConv2DBackpropFilterWithBiasAddGrad()
3594 new_node->set_assigned_device_name(orig_node->assigned_device_name()); in RewriteNode()
[all …]
Dgraph.h117 const string& assigned_device_name() const;
923 inline const string& Node::assigned_device_name() const { in assigned_device_name() function
Dsubgraph_test.cc86 EXPECT_EQ(device_info_.name(), n->assigned_device_name()) << s; in ExpectNodes()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dscheduler.cc97 curr->assigned_device_name() != out->assigned_device_name()) { in ComputeAsap()
153 src->assigned_device_name() != curr->assigned_device_name()) { in ComputeAlap()
230 Sim* sim = device_states_[event.node->assigned_device_name()]; in ComputeSchedule()
241 event.node->assigned_device_name() != out->assigned_device_name()) { in ComputeSchedule()
259 Sim* sim = device_states_[event.node->assigned_device_name()]; in ComputeSchedule()
/external/tensorflow/tensorflow/core/common_runtime/
Dplacer.cc113 node->assigned_device_name().c_str()); in LogDeviceAssignment()
116 << "): " << node->assigned_device_name(); in LogDeviceAssignment()
170 << node->assigned_device_name() << "'"; in Run()
236 if (CanAssignToDevice(input->assigned_device_name(), *devices)) { in Run()
276 CanAssignToDevice(output->assigned_device_name(), *devices)) { in Run()
Dcolocation_graph.cc446 DeviceNameUtils::ParseFullName(node.assigned_device_name(), &parsed); in AssignDevice()
453 " node's assigned device name \"", node.assigned_device_name(), in AssignDevice()
462 " node's assigned device name \"", node.assigned_device_name(), in AssignDevice()
471 "\", node's assigned device name \"", node.assigned_device_name(), in AssignDevice()
1248 &text, " framework assigned device=", node->assigned_device_name()); in DebugInfo()
1260 const string& assigned_device_name, const string& node_type, in InitializeMemberWithAssignedDevice() argument
1267 TF_RETURN_IF_ERROR(member->SetAssignedDeviceName(assigned_device_name)); in InitializeMemberWithAssignedDevice()
1271 device_set_.FindDeviceByName(assigned_device_name); in InitializeMemberWithAssignedDevice()
1276 "Assigned device '", assigned_device_name, in InitializeMemberWithAssignedDevice()
1293 return errors::Internal("Assigned device '", assigned_device_name, in InitializeMemberWithAssignedDevice()
[all …]
Dpartitioning_utils.cc34 return node->assigned_device_name(); in PartitionFunctionGraph()
Dcolocation_graph.h340 Status InitializeMemberWithAssignedDevice(const string& assigned_device_name,
Dplacer_test.cc365 EXPECT_EQ(GetNodeByName(g_, (name_a))->assigned_device_name(), \
366 GetNodeByName(g_, (name_b))->assigned_device_name()); \
372 EXPECT_NE(GetNodeByName(g_, (name_a))->assigned_device_name(), \
373 GetNodeByName(g_, (name_b))->assigned_device_name()); \
380 GetNodeByName((g), (name))->assigned_device_name()) \
387 GetNodeByName((g), (node1))->assigned_device_name()) \
392 GetNodeByName((g), (node2))->assigned_device_name()) \
398 GetNodeByName((g), (name))->assigned_device_name(), device_substr))
854 GetNodeByName(g, "in")->assigned_device_name()); in TEST_F()
Dgraph_execution_state.cc283 stateful_placements_[n->name()] = n->assigned_device_name(); in SaveStatefulNodes()
334 feed_tensor.node->assigned_device_name()); in AddNode()
/external/tensorflow/tensorflow/compiler/jit/
Dpartially_decluster_pass.cc72 DeviceNameToDeviceType(n->assigned_device_name(), &device_type)); in FindNodesToDecluster()
92 TF_RETURN_IF_ERROR(DeviceNameToDeviceType(dst->assigned_device_name(), in FindNodesToDecluster()
143 cloned_node->set_assigned_device_name(n->assigned_device_name()); in PartiallyDeclusterNode()
252 DeviceNameToDeviceType(n->assigned_device_name(), &device_type)); in MustCompileNode()
Dclone_constants_for_better_clustering.cc53 new_in->set_assigned_device_name(n->assigned_device_name()); in CloneNode()
66 DeviceNameUtils::ParseFullName(n->assigned_device_name(), &parsed)); in IsConstantOnHost()
Dbuild_xla_ops_pass.cc153 .WithAssignedDevice(new_node->assigned_device_name()), in MergeOutgoingDataEdges()
359 if (!n->assigned_device_name().empty()) { in InferDeviceForCluster()
364 device_info_cache->GetIdFor(n->assigned_device_name())); in InferDeviceForCluster()
405 DeviceNameToDeviceType(n->assigned_device_name(), &device_type)); in GetOutputMemoryTypes()
Dmark_for_compilation_pass_test_helper.cc30 if (n->assigned_device_name().empty()) { in MarkForCompilation()
Dxla_cluster_util.cc268 if (devices_seen.contains(n->assigned_device_name())) { in IsSingleGpuGraph()
272 int gpu_number = GetGpuNumber(n->assigned_device_name()); in IsSingleGpuGraph()
279 devices_seen.insert(n->assigned_device_name()); in IsSingleGpuGraph()
Dincrease_dynamism_for_auto_jit_pass.cc241 slice->assigned_device_name(), &host_name)); in ConvertTensorFlowSliceToStaticShapedSlice()
267 ops::Slice(main_scope.WithAssignedDevice(slice->assigned_device_name()) in ConvertTensorFlowSliceToStaticShapedSlice()
Dmark_for_compilation_pass.cc580 if (n.assigned_device_name().empty()) { in IgnoreResourceOpForSafetyAnalysis()
587 device_info_cache->GetCompilationDevice(n.assigned_device_name())); in IgnoreResourceOpForSafetyAnalysis()
1002 const string& device_name_str = !node->assigned_device_name().empty() in BuildInitialClusterSet()
1003 ? node->assigned_device_name() in BuildInitialClusterSet()
1162 device_info_cache_.GetDeviceTypeFor(node->assigned_device_name())); in FindCompilationCandidates()
Dnode_matchers.cc122 if (assigned_device && node->assigned_device_name() != *assigned_device) { in MatchAndExplain()
125 << " but found \"" << node->assigned_device_name() << "\""; in MatchAndExplain()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dsharding_util.cc90 string device_name = node.assigned_device_name(); in ParseShardingFromDevice()
100 string device_name = src.assigned_device_name(); in SetShardingDeviceAssignmentFromNode()
Dtf2xla_util.cc515 n->set_assigned_device_name(matching_node->assigned_device_name()); in SetNodeShardingFromNeighbors()
632 builder.Device(node->assigned_device_name().empty() in RewriteAssociatedFunction()
634 : node->assigned_device_name()); in RewriteAssociatedFunction()
/external/tensorflow/tensorflow/core/framework/
Dgraph_to_functiondef.cc171 if (!node->assigned_device_name().empty()) { in FillFunctionBody()
172 node_def->set_device(node->assigned_device_name()); in FillFunctionBody()

12