Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/core/graph/
Dgraph_partition.cc131 if (src->assigned_device_name() == dst->assigned_device_name()) { in NeedSameDeviceSendRecv()
184 builder->Attr("send_device", edge->src()->assigned_device_name()); in SetSendRecvAttrs()
187 opts.get_incarnation(edge->src()->assigned_device_name()))); in SetSendRecvAttrs()
188 builder->Attr("recv_device", edge->dst()->assigned_device_name()); in SetSendRecvAttrs()
215 cast_builder.Device(src->assigned_device_name()).Input(send_from); in AddSend()
241 send_builder.Device(src->assigned_device_name()).Input(send_from); in AddSend()
277 recv_builder.Device(dst->assigned_device_name()) in AddRecv()
290 cast_builder.Device(dst->assigned_device_name()) in AddRecv()
300 id_builder.Device(dst->assigned_device_name()) in AddRecv()
317 .Device(src->assigned_device_name()) in AddDummyConst()
[all …]
Dmkl_tfconversion_pass.cc189 conversion_node->set_assigned_device_name(src->assigned_device_name()); in InsertConversionNodeOnEdge()
274 conversion_node->set_assigned_device_name(n->assigned_device_name()); in InsertInputConversionNode()
Dmkl_layout_pass.cc807 if (!n->assigned_device_name().empty() && in CanOpRunOnCPUDevice()
808 !str_util::StrContains(n->assigned_device_name(), kCPUDeviceSubStr)) { in CanOpRunOnCPUDevice()
1640 (*out)->set_assigned_device_name(orig_node->assigned_device_name()); in GetDummyMklTensorNode()
2699 pred->assigned_device_name() != succ->assigned_device_name() || in MergeConv2DWithBiasAdd()
2803 new_node->set_assigned_device_name(pred->assigned_device_name()); in MergeConv2DWithBiasAdd()
2846 pred->assigned_device_name() != succ->assigned_device_name() || in MergePadWithConv2D()
2978 new_node->set_assigned_device_name(pred->assigned_device_name()); in MergePadWithConv2D()
3010 badd->assigned_device_name() != fltr->assigned_device_name() || in MergeConv2DBackpropFilterWithBiasAddGrad()
3100 new_node->set_assigned_device_name(badd->assigned_device_name()); in MergeConv2DBackpropFilterWithBiasAddGrad()
3225 new_node->set_assigned_device_name(orig_node->assigned_device_name()); in RewriteNode()
[all …]
Dgraph.cc118 strings::StrAppend(&ret, "{", assigned_device_name(), "}"); in DebugString()
429 copy->set_assigned_device_name(node->assigned_device_name()); in CopyNode()
651 if (!node->assigned_device_name().empty()) { in ToGraphDefSubRange()
652 node_def->set_device(node->assigned_device_name()); in ToGraphDefSubRange()
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.h115 const string& assigned_device_name() const;
889 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.cc60 node->assigned_device_name().c_str()); in LogDeviceAssignment()
63 << node->assigned_device_name(); in LogDeviceAssignment()
109 << node->assigned_device_name() << "'"; in Run()
174 if (CanAssignToDevice(input->assigned_device_name(), *devices)) { in Run()
214 CanAssignToDevice(output->assigned_device_name(), *devices)) { in Run()
Dcolocation_graph.cc412 DeviceNameUtils::ParseFullName(node.assigned_device_name(), &parsed); in AssignDevice()
420 " node's assigned device name \"", node.assigned_device_name(), in AssignDevice()
430 "\", node's assigned device name \"", node.assigned_device_name(), in AssignDevice()
909 const string& assigned_device_name, const string& node_type, in InitializeMemberWithAssignedDevice() argument
916 TF_RETURN_IF_ERROR(member->SetAssignedDeviceName(assigned_device_name)); in InitializeMemberWithAssignedDevice()
922 device_set_->FindDeviceByName(assigned_device_name); in InitializeMemberWithAssignedDevice()
924 return errors::Internal("Assigned device '", assigned_device_name, in InitializeMemberWithAssignedDevice()
934 return errors::Internal("Assigned device '", assigned_device_name, in InitializeMemberWithAssignedDevice()
945 node.assigned_device_name(), node.type_string(), true, member)); in InitializeMember()
Dcolocation_graph.h231 Status InitializeMemberWithAssignedDevice(const string& assigned_device_name,
Dplacer_test.cc292 EXPECT_EQ(GetNodeByName(g_, (name_a))->assigned_device_name(), \
293 GetNodeByName(g_, (name_b))->assigned_device_name()); \
299 EXPECT_NE(GetNodeByName(g_, (name_a))->assigned_device_name(), \
300 GetNodeByName(g_, (name_b))->assigned_device_name()); \
307 GetNodeByName((g), (name))->assigned_device_name()) \
313 GetNodeByName((g), (name))->assigned_device_name(), device_substr))
742 GetNodeByName(g, "in")->assigned_device_name()); in TEST_F()
Dpartitioning_utils.cc34 return node->assigned_device_name(); in PartitionFunctionGraph()
/external/tensorflow/tensorflow/compiler/jit/
Dpartially_decluster_pass.cc60 DeviceToDeviceType(n->assigned_device_name(), &device_type)); in FindNodesToDecluster()
81 DeviceToDeviceType(dst->assigned_device_name(), &dst_device_type)); in FindNodesToDecluster()
131 cloned_node->set_assigned_device_name(n->assigned_device_name()); in PartiallyDeclusterNode()
240 DeviceToDeviceType(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()
67 DeviceNameUtils::ParseFullName(n->assigned_device_name(), &parsed)); in IsSmallHostConstant()
Dxla_fusion_optimizer.cc276 VLOG(2) << "Device names " << node_from->assigned_device_name() << " " in Optimize()
277 << node_to->assigned_device_name(); in Optimize()
Dmark_for_compilation_pass_test_helper.cc28 if (n->assigned_device_name().empty()) { in MarkForCompilation()
Dmark_for_compilation_pass.cc492 DeviceToDeviceType(node->assigned_device_name(), &device_type)); in FindCompilationCandidates()
927 if (n.assigned_device_name().empty()) { in IgnoreResourceOpForSafetyAnalysis()
933 DeviceToDeviceType(n.assigned_device_name(), &device_type)); in IgnoreResourceOpForSafetyAnalysis()
1116 const string& device = !node->assigned_device_name().empty() in RunImpl()
1117 ? node->assigned_device_name() in RunImpl()
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()
Dbuild_xla_ops_pass.cc292 if (!n->assigned_device_name().empty()) { in InferDeviceForCluster()
296 device_names.insert(n->assigned_device_name()); in InferDeviceForCluster()
Dnode_matchers.cc118 if (assigned_device && node->assigned_device_name() != *assigned_device) { in MatchAndExplain()
121 << " 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.cc522 n->set_assigned_device_name(matching_node->assigned_device_name()); in SetNodeShardingFromNeighbors()
639 builder.Device(node->assigned_device_name().empty() in RewriteAssociatedFunction()
641 : node->assigned_device_name()); in RewriteAssociatedFunction()
/external/tensorflow/tensorflow/core/framework/
Dgraph_to_functiondef.cc164 if (!node->assigned_device_name().empty()) { in GraphToFunctionDef()
165 node_def->set_device(node->assigned_device_name()); in GraphToFunctionDef()
/external/tensorflow/tensorflow/c/
Dc_api_function.cc176 if (!node->assigned_device_name().empty()) { in FillFunctionBody()
177 node_def->set_device(node->assigned_device_name()); in FillFunctionBody()

12