Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/core/common_runtime/
Dreplicate_per_replica_nodes.cc104 const string& dst_device = dst->assigned_device_name(); in ReplicateFromCompositeDeviceToRegularDevice()
107 if (replicated_node->assigned_device_name() == dst_device) { in ReplicateFromCompositeDeviceToRegularDevice()
143 pack_node->set_assigned_device_name(dst->assigned_device_name()); in ReplicateFromCompositeDeviceToRegularDevice()
154 edge->src()->assigned_device_name(), " to node ", dst->name(), in ReplicateFromCompositeDeviceToRegularDevice()
194 if (src->assigned_device_name() != node->assigned_device_name()) { in ReplicateEdges()
207 if (dst->assigned_device_name() != node->assigned_device_name()) { in ReplicateEdges()
233 if (composite_device_names.find(n->assigned_device_name()) != in ReplicatePerReplicaNodesInFunctionGraph()
238 composite_device_to_cluster_nodes[n->assigned_device_name()].push_back(n); in ReplicatePerReplicaNodesInFunctionGraph()
Dlower_while_op_test.cc188 const string assigned_device_name = "/job:localhost/replica:0/task:0/gpu:0"; in TEST() local
189 placeholder->set_assigned_device_name(assigned_device_name); in TEST()
221 ASSERT_EQ(consumer->assigned_device_name(), assigned_device_name); in TEST()
232 ASSERT_EQ(consumer->assigned_device_name(), assigned_device_name); in TEST()
241 ASSERT_EQ(consumer->assigned_device_name(), assigned_device_name); in TEST()
252 ASSERT_EQ(consumer->assigned_device_name(), assigned_device_name); in TEST()
261 ASSERT_EQ(consumer->assigned_device_name(), assigned_device_name); in TEST()
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()
Dlower_while_op.cc248 .AssignedDevice(edge->src()->assigned_device_name()); in CreateEnterNodes()
288 .AssignedDevice(enter_node->assigned_device_name()) in CreateMergeNodes()
340 .AssignedDevice(merge_node->assigned_device_name()) in CreateSwitchNodes()
399 ->assigned_device_name()) in CreateExitNodes()
453 .AssignedDevice(merge_node->assigned_device_name()) in CreateNextIterationNodes()
Dcolocation_graph.cc473 DeviceNameUtils::ParseFullName(node.assigned_device_name(), &parsed); in AssignDevice()
480 " node's assigned device name \"", node.assigned_device_name(), in AssignDevice()
489 " node's assigned device name \"", node.assigned_device_name(), in AssignDevice()
498 "\", node's assigned device name \"", node.assigned_device_name(), in AssignDevice()
1377 &text, " framework assigned device=", node->assigned_device_name()); in DebugInfo()
1389 const string& assigned_device_name, const string& node_type, in InitializeMemberWithAssignedDevice() argument
1396 TF_RETURN_IF_ERROR(member->SetAssignedDeviceName(assigned_device_name)); in InitializeMemberWithAssignedDevice()
1400 device_set_.FindDeviceByName(assigned_device_name); in InitializeMemberWithAssignedDevice()
1405 "Assigned device '", assigned_device_name, in InitializeMemberWithAssignedDevice()
1421 return errors::Internal("Assigned device '", assigned_device_name, in InitializeMemberWithAssignedDevice()
[all …]
Dmkl_tfconversion_pass.cc190 conversion_node->set_assigned_device_name(src->assigned_device_name()); in InsertConversionNodeOnEdge()
276 conversion_node->set_assigned_device_name(n->assigned_device_name()); in InsertInputConversionNode()
Dfunction_utils.cc321 if (!n->assigned_device_name().empty()) { in ToGraphDef()
322 ndef->set_device(n->assigned_device_name()); in ToGraphDef()
Dmkl_layout_pass.cc1066 if (!n->assigned_device_name().empty() && in CanOpRunOnCPUDevice()
1067 !absl::StrContains(n->assigned_device_name(), kCPUDeviceSubStr)) { in CanOpRunOnCPUDevice()
2150 (*out)->set_assigned_device_name(orig_node->assigned_device_name()); in GetDummyMklTensorNode()
3003 pred->assigned_device_name() != succ->assigned_device_name() || in MergeConv2DWithBiasAdd()
3121 new_node->set_assigned_device_name(pred->assigned_device_name()); in MergeConv2DWithBiasAdd()
3164 pred->assigned_device_name() != succ->assigned_device_name() || in MergePadWithConv2D()
3297 new_node->set_assigned_device_name(pred->assigned_device_name()); in MergePadWithConv2D()
3329 badd->assigned_device_name() != fltr->assigned_device_name() || in MergeConv2DBackpropFilterWithBiasAddGrad()
3431 new_node->set_assigned_device_name(badd->assigned_device_name()); in MergeConv2DBackpropFilterWithBiasAddGrad()
3667 new_node->set_assigned_device_name(orig_node->assigned_device_name()); in RewriteNode()
[all …]
Dcolocation_graph.h352 Status InitializeMemberWithAssignedDevice(const string& assigned_device_name,
Dpartitioning_utils.cc37 return node->assigned_device_name(); in PartitionFunctionGraph()
Dplacer_test.cc368 EXPECT_EQ(GetNodeByName(g_, (name_a))->assigned_device_name(), \
369 GetNodeByName(g_, (name_b))->assigned_device_name()); \
375 EXPECT_NE(GetNodeByName(g_, (name_a))->assigned_device_name(), \
376 GetNodeByName(g_, (name_b))->assigned_device_name()); \
383 GetNodeByName((g), (name))->assigned_device_name()) \
390 GetNodeByName((g), (node1))->assigned_device_name()) \
395 GetNodeByName((g), (node2))->assigned_device_name()) \
401 GetNodeByName((g), (name))->assigned_device_name(), device_substr))
857 GetNodeByName(g, "in")->assigned_device_name()); in TEST_F()
/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()
219 cast_builder.Device(src->assigned_device_name()).Input(send_from); in AddSend()
245 send_builder.Device(src->assigned_device_name()).Input(send_from); in AddSend()
284 << " on " << src->assigned_device_name() << " in " in AddRecv()
287 << " on " << dst->assigned_device_name() << " in " in AddRecv()
294 << " on " << src->assigned_device_name() << " for " << dst->name() in AddRecv()
296 << " on " << dst->assigned_device_name(); in AddRecv()
[all …]
Dcostmodel.cc372 if (persistent_alloc_ids_by_devices_.find(node->assigned_device_name()) == in IsPersistentTensor()
376 return persistent_alloc_ids_by_devices_.at(node->assigned_device_name()) in IsPersistentTensor()
488 cnode->set_device(n->assigned_device_name()); in AddToCostGraphDef()
Dgraph.cc115 strings::StrAppend(&ret, "{", assigned_device_name(), "}"); in DebugString()
477 copy->set_assigned_device_name(node->assigned_device_name()); in CopyNode()
705 if (!node->assigned_device_name().empty()) { in ToGraphDefSubRange()
706 node_def->set_device(node->assigned_device_name()); in ToGraphDefSubRange()
/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/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()
Dmark_for_compilation_pass_test_helper.cc30 if (n->assigned_device_name().empty()) { in MarkForCompilation()
Dbuild_xla_ops_pass.cc153 .WithAssignedDevice(new_node->assigned_device_name()), in MergeOutgoingDataEdges()
363 if (!n->assigned_device_name().empty()) { in InferDeviceForCluster()
368 device_info_cache->GetIdFor(n->assigned_device_name())); in InferDeviceForCluster()
409 DeviceNameToDeviceType(n->assigned_device_name(), &device_type)); in GetOutputMemoryTypes()
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.cc589 if (n.assigned_device_name().empty()) { in IgnoreResourceOpForSafetyAnalysis()
596 device_info_cache->GetCompilationDevice(n.assigned_device_name())); in IgnoreResourceOpForSafetyAnalysis()
1027 const string& device_name_str = !node->assigned_device_name().empty() in BuildInitialClusterSet()
1028 ? node->assigned_device_name() in BuildInitialClusterSet()
1181 device_info_cache_.GetDeviceTypeFor(node->assigned_device_name())); in FindCompilationCandidates()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dsharding_util.cc102 string device_name = node.assigned_device_name(); in ParseShardingFromDevice()
140 string device_name = src.assigned_device_name(); in SetShardingDeviceAssignmentFromNode()
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dvariable_merger_pass.cc115 node->set_assigned_device_name(handle_op->assigned_device_name()); in MergeReadVariableOps()
150 var_handle_ops_by_device[m->assigned_device_name()].push_back(m); in Run()
Ddistributed_tpu_rewrite_pass.cc673 paddings_def.set_device(split_node->assigned_device_name()); in CreatePadNode()
695 pad_def.set_device(split_node->assigned_device_name()); in CreatePadNode()
701 pad_node->set_assigned_device_name(split_node->assigned_device_name()); in CreatePadNode()
718 const std::string input_assigned_device = orig_src->assigned_device_name(); in CreateSplitNode()
1084 const string& assigned_device = unpadded_input->assigned_device_name(); in SetPaddingNodesDevices()
1112 if (!node->assigned_device_name().empty()) { in AssignedOrRequestedDevice()
1113 return node->assigned_device_name(); in AssignedOrRequestedDevice()
1172 node->set_assigned_device_name(src_node->assigned_device_name()); in PlaceOpsOnTPU()
2285 def.set_device(src->assigned_device_name()); in BuildDynamicShapeNodes()
2295 shape_node->set_assigned_device_name(src->assigned_device_name()); in BuildDynamicShapeNodes()
[all …]

12