Home
last modified time | relevance | path

Searched refs:dst_node (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/graph/
Dcollective_order.cc101 Node* dst_node = collective_nodes[dst_idx]; in CreateControlDependencies() local
104 << dst_node->name() << " instance " << instance_keys[dst_idx]; in CreateControlDependencies()
105 (*dependency_edges)[src_node].insert(dst_node); in CreateControlDependencies()
107 src_paths.insert(dst_node); in CreateControlDependencies()
108 for (Node* downstream_node : all_paths[dst_node]) { in CreateControlDependencies()
154 for (Node* dst_node : pair.second) { in InsertControlDependencies()
155 graph->AddControlEdge(src_node, dst_node); in InsertControlDependencies()
166 for (Node* dst_node : pair.second) { in InsertControlDependencies()
167 wait_for[dst_node].insert(src_instance); in InsertControlDependencies()
Dgraph_constructor.cc315 : src_name(name), src_index(i1), dst_node(n), dst_index(i2) {} in EdgeInfo()
319 Node* dst_node; member
1072 g_->AddControlEdge(src_node, e.dst_node); in AddBackEdges()
1075 MakeEdge(src_node, e.src_index, e.dst_node, e.dst_index)); in AddBackEdges()
1079 << e.dst_node->name(); in AddBackEdges()
/external/tensorflow/tensorflow/core/grappler/utils/
Dgrappler_test.cc178 const NodeDef* dst_node = node_map.GetNode(dst); in IsNodesDirectlyConnected() local
180 EXPECT_TRUE(dst_node != nullptr) << dst << " node not found"; in IsNodesDirectlyConnected()
181 return src_node && dst_node && dst_node->input(position) == src_node->name(); in IsNodesDirectlyConnected()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dfunctionalize_cond.cc1193 Node* dst_node = oe->dst(); in RemoveRedundantMerge() local
1200 dst_node, dst_port); in RemoveRedundantMerge()
1246 Node* dst_node = e->dst(); in RemoveRedundantSwitch() local
1251 if (IsMerge(dst_node)) { in RemoveRedundantSwitch()
1252 auto id_or = JoinCondStatesMerge(dst_node, dst_id, in RemoveRedundantSwitch()
1253 state_map_.LookupCondId(dst_node)); in RemoveRedundantSwitch()
1255 FormatNodeForError(*dst_node)); in RemoveRedundantSwitch()
1256 state_map_.ResetCondId(dst_node, id_or.ValueOrDie()); in RemoveRedundantSwitch()
1259 JoinCondStatesNonMerge(dst_id, state_map_.LookupCondId(dst_node)); in RemoveRedundantSwitch()
1261 state_map_.ResetCondId(dst_node, id_or.ValueOrDie()); in RemoveRedundantSwitch()
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dremote_fused_graph_execute_utils.cc796 const Node* dst_node = out_edge->dst(); in ClusterizeNodes() local
797 CHECK_NOTNULL(dst_node); in ClusterizeNodes()
798 const bool dst_is_outside = node_names.count(dst_node->name()) <= 0; in ClusterizeNodes()
802 if (dst_node->IsSink()) { in ClusterizeNodes()
982 Node* dst_node = edge->dst(); in FuseCluster() local
987 graph.AddEdge(fused_node, j, dst_node, dst_input); in FuseCluster()
/external/tensorflow/tensorflow/core/common_runtime/
Dexecutor.cc2592 const Node* dst_node = e->dst(); in DeleteFrame() local
2595 impl_->gview_.node(dst_node->id())->pending_id; in DeleteFrame()
2599 if (dst_node->IsSink()) continue; in DeleteFrame()
2604 if (IsMerge(dst_node)) { in DeleteFrame()
2609 dst_dead = (dead_cnt == dst_node->num_inputs()); in DeleteFrame()
2614 dst_dead = (dead_cnt == dst_node->num_inputs()); in DeleteFrame()
2624 if (IsControlTrigger(dst_node)) dst_dead = false; in DeleteFrame()
2625 ready->emplace_back(dst_node, parent_frame, parent_iter, dst_dead); in DeleteFrame()
/external/tensorflow/tensorflow/compiler/jit/
Dencapsulate_subgraphs_pass.cc797 Node* dst_node = edge->dst(); in RecordArg() local
798 Node* dst_image = node_images.at(dst_node); in RecordArg()
800 args_by_dst_[InputTensor(dst_node, dst_slot)] = arg_index; in RecordArg()
1025 const Node* dst_node = output.first.node; in AddHostComputes() local
1026 Node* dst_image = node_images.at(dst_node); in AddHostComputes()
1035 for (const auto& dst_node : oc_subgraph.control_outputs) { in AddHostComputes() local
1036 Node* dst_image = node_images.at(dst_node); in AddHostComputes()
/external/vulkan-validation-layers/layers/
Ddescriptor_sets.cpp1743 auto dst_node = GetSetNode(dst_set); in ValidateUpdateDescriptorSets() local
1746 assert(dst_node); in ValidateUpdateDescriptorSets()
1749 …if (!dst_node->ValidateCopyUpdate(report_data, &p_cds[i], src_node, func_name, &error_code, &error… in ValidateUpdateDescriptorSets()
1780 auto dst_node = dev_data->GetSetNode(dst_set); in PerformUpdateDescriptorSets() local
1781 if (src_node && dst_node) { in PerformUpdateDescriptorSets()
1782 dst_node->PerformCopyUpdate(&p_cds[i], src_node); in PerformUpdateDescriptorSets()