Home
last modified time | relevance | path

Searched refs:src_node (Results 1 – 22 of 22) sorted by relevance

/external/tensorflow/tensorflow/core/graph/
Dnode_builder.cc51 NodeBuilder& NodeBuilder::Input(Node* src_node, int src_index) { in Input() argument
52 inputs_.emplace_back(src_node, src_index); in Input()
54 if (GetOutputType(src_node, src_index, &dt)) { in Input()
55 def_builder_.Input(src_node->name(), src_index, dt); in Input()
85 NodeBuilder& NodeBuilder::ControlInput(Node* src_node) { in ControlInput() argument
86 control_inputs_.emplace_back(src_node); in ControlInput()
87 def_builder_.ControlInput(src_node->name()); in ControlInput()
94 for (const Node* src_node : src_nodes) { in ControlInputs() local
95 def_builder_.ControlInput(src_node->name()); in ControlInputs()
Dcontrol_flow.cc94 const Node* src_node = g->source_node(); in BuildControlFlowInfo() local
95 ControlFlowInfo& src_info = (*info)[src_node->id()]; in BuildControlFlowInfo()
96 src_info.frame = src_node; in BuildControlFlowInfo()
97 src_info.parent_frame = src_node; in BuildControlFlowInfo()
101 ready.push_back(src_node); in BuildControlFlowInfo()
Dcollective_order.cc100 Node* src_node = collective_nodes[src_idx]; in CreateControlDependencies() local
102 VLOG(1) << "Adding control dependency from node " << src_node->name() in CreateControlDependencies()
105 (*dependency_edges)[src_node].insert(dst_node); in CreateControlDependencies()
106 auto& src_paths = all_paths[src_node]; in CreateControlDependencies()
153 Node* src_node = pair.first; in InsertControlDependencies() local
155 graph->AddControlEdge(src_node, dst_node); in InsertControlDependencies()
Dgraph_constructor.cc982 Node* src_node; in Convert() local
989 src_node = iter->second.node; in Convert()
991 if (src_node == nullptr) has_data_back_edge = true; in Convert()
996 src_node = iter->second; in Convert()
1000 if (src_node != nullptr && src_index >= src_node->num_outputs()) { in Convert()
1004 src_node->num_outputs(), " outputs"); in Convert()
1007 inputs.emplace_back(string(id.first), src_node, src_index); in Convert()
1070 Node* src_node = gdef_nodes_[e.src_name].node; in AddBackEdges() local
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()
[all …]
Dnode_builder.h91 NodeBuilder& Input(Node* src_node, int src_index = 0);
98 NodeBuilder& ControlInput(Node* src_node);
/external/tensorflow/tensorflow/core/framework/
Dnode_def_builder.cc96 NodeDefBuilder& NodeDefBuilder::Input(StringPiece src_node, int src_index, in Input() argument
99 if (arg != nullptr) SingleInput(arg, src_node, src_index, dt); in Input()
116 StringPiece src_node, int src_index, in SingleInput() argument
118 AddInput(src_node, src_index); in SingleInput()
173 void NodeDefBuilder::AddInput(StringPiece src_node, int src_index) { in AddInput() argument
174 if (src_node.empty()) { in AddInput()
176 } else if (src_node[0] == '^') { in AddInput()
178 strings::StrCat("Non-control input starting with ^: ", src_node)); in AddInput()
180 node_def_.add_input(strings::StrCat(src_node, ":", src_index)); in AddInput()
182 node_def_.add_input(string(src_node)); in AddInput()
[all …]
Dnode_def_builder.h78 NodeDefBuilder& Input(StringPiece src_node, int src_index, DataType dt);
88 NodeDefBuilder& ControlInput(StringPiece src_node);
153 void SingleInput(const OpDef::ArgDef* input_arg, StringPiece src_node,
159 void AddInput(StringPiece src_node, int src_index);
/external/tensorflow/tensorflow/core/debug/
Ddebug_graph_utils.cc106 for (Node* src_node : graph->nodes()) { in InsertNodes()
109 for (const Edge* edge : src_node->out_edges()) { in InsertNodes()
119 for (int src_output_slot = 0; src_output_slot < src_node->num_outputs(); in InsertNodes()
122 strings::StrCat(src_node->name(), ":", src_output_slot); in InsertNodes()
143 const DataType src_dt = src_node->output_type(src_output_slot); in InsertNodes()
145 TF_RETURN_IF_ERROR(MemoryTypeForOutput(device_type, graph, src_node, in InsertNodes()
151 graph, device_type, memory_type == HOST_MEMORY, src_node->name(), in InsertNodes()
162 graph->AddEdge(src_node, src_output_slot, copy_node, 0); in InsertNodes()
192 const bool is_ref = IsRefType(src_node->output_type(src_output_slot)); in InsertNodes()
206 if (!src_node->IsEnter() && !src_node->IsNextIteration()) { in InsertNodes()
/external/tensorflow/tensorflow/compiler/jit/
Dencapsulate_subgraphs_pass.cc773 Node* src_node = edge->src(); in RecordArg() local
778 OutputTensor(src_node, src_slot), args_by_src_.size()); in RecordArg()
783 absl::StrCat(src_node->name(), "_", src_slot, "_arg"), kArgOp, in RecordArg()
784 NodeDebugInfo(src_node->def())); in RecordArg()
794 src_arg_pairs->push_back({src_node, arg}); in RecordArg()
808 Node* src_node = edge->src(); in RecordResult() local
809 Node* src_image = node_images.at(src_node); in RecordResult()
814 results_.emplace(OutputTensor(src_node, src_slot), results_.size()); in RecordResult()
819 absl::StrCat(src_node->name(), "_", src_slot, "_retval"), kRetValOp, in RecordResult()
820 NodeDebugInfo(src_node->def())); in RecordResult()
[all …]
/external/tensorflow/tensorflow/core/grappler/utils/
Dgrappler_test.cc177 const NodeDef* src_node = node_map.GetNode(src); in IsNodesDirectlyConnected() local
179 EXPECT_TRUE(src_node != nullptr) << src << " node not found"; in IsNodesDirectlyConnected()
181 return src_node && dst_node && dst_node->input(position) == src_node->name(); in IsNodesDirectlyConnected()
/external/tensorflow/tensorflow/core/nccl/
Dnccl_rewrite.cc61 Node* src_node = edge->src(); in ReplaceReduce() local
65 int send_dev = src_node->assigned_device_name_index(); in ReplaceReduce()
67 recv_builder.Input(src_node); in ReplaceReduce()
73 .Input(src_node) in ReplaceReduce()
/external/tensorflow/tensorflow/core/common_runtime/
Dprocess_function_library_runtime.cc330 Node* src_node = it->src(); in PinArgsAndRets() local
331 const string* src_device = AssignedOrRequestedDeviceName(*src_node); in PinArgsAndRets()
333 GetColocationGroup(src_node, &colocation_group); in PinArgsAndRets()
334 VLOG(3) << "Considering src: " << src_node->name() in PinArgsAndRets()
338 src_node->IsIdentity()) { in PinArgsAndRets()
339 src_node = *src_node->in_nodes().begin(); in PinArgsAndRets()
340 src_device = AssignedOrRequestedDeviceName(*src_node); in PinArgsAndRets()
341 GetColocationGroup(src_node, &colocation_group); in PinArgsAndRets()
342 VLOG(3) << "Considering src: " << src_node->name() in PinArgsAndRets()
/external/tensorflow/tensorflow/core/kernels/
Dremote_fused_graph_execute_utils.cc774 const Node* src_node = in_edge->src(); in ClusterizeNodes() local
776 node_names.count(src_node->name()) <= 0 && !src_node->IsSource(); in ClusterizeNodes()
779 strings::StrCat(src_node->name(), ":", in_edge->src_output()); in ClusterizeNodes()
780 CHECK_EQ(1, src_node->num_outputs()) in ClusterizeNodes()
782 << src_node->name(); in ClusterizeNodes()
912 const Node* src_node = edge->src(); in BuildClusterByBorder() local
913 CHECK_NOTNULL(src_node); in BuildClusterByBorder()
918 if (tid.first == src_node->name() && tid.second == src_port) { in BuildClusterByBorder()
920 border_input_nodes.insert(src_node); in BuildClusterByBorder()
923 if (visited.insert(src_node).second) { in BuildClusterByBorder()
[all …]
/external/mesa3d/src/compiler/nir/
Dnir_from_ssa.c386 merge_node *src_node = get_merge_node(src->src.ssa, state); in coalesce_phi_nodes_block() local
387 if (src_node->set != dest_node->set) in coalesce_phi_nodes_block()
388 merge_merge_sets(dest_node->set, src_node->set); in coalesce_phi_nodes_block()
413 merge_node *src_node = get_merge_node(entry->src.ssa, state); in aggressive_coalesce_parallel_copy() local
416 if (src_node->set == dest_node->set) in aggressive_coalesce_parallel_copy()
419 if (!merge_sets_interfere(src_node->set, dest_node->set)) in aggressive_coalesce_parallel_copy()
420 merge_merge_sets(src_node->set, dest_node->set); in aggressive_coalesce_parallel_copy()
/external/tensorflow/tensorflow/lite/toco/tensorflow_graph_matching/
Dresolve_cluster.cc114 for (const std::unique_ptr<tensorflow::NodeDef>& src_node : in MaybeResolveClusters() local
117 AddNodeToGraph(*src_node, cluster_names, pruned_graph.get()); in MaybeResolveClusters()
/external/e2fsprogs/lib/ext2fs/
Dblkmap64_rb.c244 struct rb_node *dest_node, *src_node, *dest_last, **n; in rb_copy_bmap() local
256 src_node = ext2fs_rb_first(&src_bp->root); in rb_copy_bmap()
257 while (src_node) { in rb_copy_bmap()
258 src_ext = node_to_extent(src_node); in rb_copy_bmap()
278 src_node = ext2fs_rb_next(src_node); in rb_copy_bmap()
/external/u-boot/lib/
Dfdtdec.c721 int fdtdec_parse_phandle_with_args(const void *blob, int src_node, in fdtdec_parse_phandle_with_args() argument
734 list = fdt_getprop(blob, src_node, list_name, &size); in fdtdec_parse_phandle_with_args()
764 fdt_get_name(blob, src_node, in fdtdec_parse_phandle_with_args()
775 fdt_get_name(blob, src_node, in fdtdec_parse_phandle_with_args()
792 fdt_get_name(blob, src_node, NULL)); in fdtdec_parse_phandle_with_args()
813 fdt_get_name(blob, src_node, in fdtdec_parse_phandle_with_args()
/external/u-boot/include/
Dfdtdec.h215 int fdtdec_parse_phandle_with_args(const void *blob, int src_node,
/external/vulkan-validation-layers/layers/
Ddescriptor_sets.cpp1742 auto src_node = GetSetNode(src_set); in ValidateUpdateDescriptorSets() local
1745 assert(src_node); in ValidateUpdateDescriptorSets()
1749 …if (!dst_node->ValidateCopyUpdate(report_data, &p_cds[i], src_node, func_name, &error_code, &error… in ValidateUpdateDescriptorSets()
1779 auto src_node = dev_data->GetSetNode(src_set); in PerformUpdateDescriptorSets() local
1781 if (src_node && dst_node) { in PerformUpdateDescriptorSets()
1782 dst_node->PerformCopyUpdate(&p_cds[i], src_node); in PerformUpdateDescriptorSets()
/external/expat/lib/
Dxmlparse.c7114 int src_node, in build_node() argument
7120 dest->type = dtd->scaffold[src_node].type; in build_node()
7121 dest->quant = dtd->scaffold[src_node].quant; in build_node()
7125 src = dtd->scaffold[src_node].name; in build_node()
7138 dest->numchildren = dtd->scaffold[src_node].childcnt; in build_node()
7141 for (i = 0, cn = dtd->scaffold[src_node].firstchild; in build_node()
/external/python/cpython3/Modules/expat/
Dxmlparse.c7114 int src_node, in build_node() argument
7120 dest->type = dtd->scaffold[src_node].type; in build_node()
7121 dest->quant = dtd->scaffold[src_node].quant; in build_node()
7125 src = dtd->scaffold[src_node].name; in build_node()
7138 dest->numchildren = dtd->scaffold[src_node].childcnt; in build_node()
7141 for (i = 0, cn = dtd->scaffold[src_node].firstchild; in build_node()
/external/python/cpython2/Modules/expat/
Dxmlparse.c7114 int src_node, in build_node() argument
7120 dest->type = dtd->scaffold[src_node].type; in build_node()
7121 dest->quant = dtd->scaffold[src_node].quant; in build_node()
7125 src = dtd->scaffold[src_node].name; in build_node()
7138 dest->numchildren = dtd->scaffold[src_node].childcnt; in build_node()
7141 for (i = 0, cn = dtd->scaffold[src_node].firstchild; in build_node()