Home
last modified time | relevance | path

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

123

/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/onnx/pattern_conversion/
H A Dcommon.cpp31 auto src_node = node->input(0)->node(); in FetchSliceAndSelect() local
32 while (src_node) { in FetchSliceAndSelect()
33 if ((src_node->kind() == aten::slice || src_node->kind() == aten::select) && in FetchSliceAndSelect()
34 IsSameSource(src_node, node)) { in FetchSliceAndSelect()
35 slice_and_select_node.emplace_back(src_node); in FetchSliceAndSelect()
36 src_node = src_node->input(0)->node(); in FetchSliceAndSelect()
38 src_node = nullptr; in FetchSliceAndSelect()
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A 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 …]
H A Dnode_def_builder.h80 NodeDefBuilder& Input(StringPiece src_node, int src_index, DataType dt);
90 NodeDefBuilder& ControlInput(StringPiece src_node);
159 void SingleInput(const OpDef::ArgDef* input_arg, StringPiece src_node,
165 void AddInput(StringPiece src_node, int src_index);
/aosp_15_r20/external/tensorflow/tensorflow/core/graph/
H A Dnode_builder.cc56 NodeBuilder& NodeBuilder::Input(Node* src_node, int src_index) { in Input() argument
57 inputs_.emplace_back(src_node, src_index); in Input()
59 if (GetOutputType(src_node, src_index, &dt)) { in Input()
60 def_builder_.Input(src_node->name(), src_index, dt); in Input()
90 NodeBuilder& NodeBuilder::ControlInput(Node* src_node) { in ControlInput() argument
91 control_inputs_.emplace_back(src_node); in ControlInput()
92 def_builder_.ControlInput(src_node->name()); in ControlInput()
99 for (const Node* src_node : src_nodes) { in ControlInputs() local
100 def_builder_.ControlInput(src_node->name()); in ControlInputs()
H A 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()
H A 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()
H A Dnode_builder.h91 NodeBuilder& Input(Node* src_node, int src_index = 0);
98 NodeBuilder& ControlInput(Node* src_node);
/aosp_15_r20/external/tensorflow/tensorflow/core/debug/
H A Ddebug_graph_utils.cc133 for (Node* src_node : graph->nodes()) { in InsertNodes()
136 for (const Edge* edge : src_node->out_edges()) { in InsertNodes()
146 for (int src_output_slot = 0; src_output_slot < src_node->num_outputs(); in InsertNodes()
149 strings::StrCat(src_node->name(), ":", src_output_slot); in InsertNodes()
170 const DataType src_dt = src_node->output_type(src_output_slot); in InsertNodes()
172 TF_RETURN_IF_ERROR(MemoryTypeForOutput(device_type, graph, src_node, in InsertNodes()
185 src_node->name(), src_output_slot, src_dt, tensor_name, in InsertNodes()
195 graph->AddEdge(src_node, src_output_slot, copy_node, 0); in InsertNodes()
225 const bool is_ref = IsRefType(src_node->output_type(src_output_slot)); in InsertNodes()
239 if (!src_node->IsEnter() && !src_node->IsNextIteration()) { in InsertNodes()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/
H A Dencapsulate_subgraphs_pass.cc452 Node* src_node = edge->src(); in RecordArg() local
457 OutputTensor(src_node, src_slot), args_by_src_.size()); in RecordArg()
462 absl::StrCat(src_node->name(), "_", src_slot, "_arg"), kArgOp, in RecordArg()
463 NodeDebugInfo(src_node->def())); in RecordArg()
471 src_arg_pairs->push_back({src_node, arg}); in RecordArg()
485 Node* src_node = edge->src(); in RecordControlResult() local
486 Node* src_image = node_images.at(src_node); in RecordControlResult()
494 Node* src_node = edge->src(); in RecordResult() local
495 Node* src_image = node_images.at(src_node); in RecordResult()
500 results_.emplace(OutputTensor(src_node, src_slot), results_.size()); in RecordResult()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dgraph_constructor.cc1181 Node* src_node; in Convert() local
1188 src_node = iter->second.node; in Convert()
1190 if (src_node == nullptr) has_data_back_edge = true; in Convert()
1195 src_node = iter->second; in Convert()
1199 if (src_node != nullptr && src_index >= src_node->num_outputs()) { in Convert()
1203 << " which has " << src_node->num_outputs() << " outputs."; in Convert()
1205 if (src_node->type_string() == "If" || in Convert()
1206 src_node->type_string() == "StatelessIf" || in Convert()
1207 src_node->type_string() == "While" || in Convert()
1208 src_node->type_string() == "StatelessWhile") { in Convert()
[all …]
H A Dprocess_function_library_runtime.cc489 Node* src_node = it->src(); in PinArgsAndRets() local
490 const string* src_device = AssignedOrRequestedDeviceName(*src_node); in PinArgsAndRets()
492 GetColocationGroup(src_node, &colocation_group); in PinArgsAndRets()
493 VLOG(3) << "Considering src: " << src_node->name() in PinArgsAndRets()
497 src_node->IsIdentity()) { in PinArgsAndRets()
500 TF_RETURN_IF_ERROR(src_node->input_node(0, &input_node)); in PinArgsAndRets()
501 src_node = input_node; in PinArgsAndRets()
503 src_device = AssignedOrRequestedDeviceName(*src_node); in PinArgsAndRets()
504 GetColocationGroup(src_node, &colocation_group); in PinArgsAndRets()
505 VLOG(3) << "Considering src: " << src_node->name() in PinArgsAndRets()
[all …]
/aosp_15_r20/external/mesa3d/src/compiler/nir/
H A Dnir_from_ssa.c455 merge_node *src_node = get_merge_node(src->src.ssa, state); in coalesce_phi_nodes_block() local
456 if (src_node->set != dest_node->set) in coalesce_phi_nodes_block()
457 merge_merge_sets(dest_node->set, src_node->set); in coalesce_phi_nodes_block()
480 merge_node *src_node = get_merge_node(entry->src.ssa, state); in aggressive_coalesce_parallel_copy() local
483 if (src_node->set == dest_node->set) in aggressive_coalesce_parallel_copy()
489 if (dest_node->set->divergent != src_node->set->divergent) in aggressive_coalesce_parallel_copy()
492 if (!merge_sets_interfere(src_node->set, dest_node->set)) in aggressive_coalesce_parallel_copy()
493 merge_merge_sets(src_node->set, dest_node->set); in aggressive_coalesce_parallel_copy()
645 merge_node *src_node = (merge_node *)entry->data; in remove_no_op_phi() local
646 assert(src_node->set == node->set); in remove_no_op_phi()
H A Dnir_lower_io_to_temporaries.c55 foreach_two_lists(dest_node, dest_vars, src_node, src_vars) { in emit_copies()
57 nir_variable *src = exec_node_data(nir_variable, src_node, node); in emit_copies()
/aosp_15_r20/external/tensorflow/tensorflow/core/nccl/
H A 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()
H A Dnccl_manager_test.cc211 TensorShape shape, int src_node, int src_rank, in MakeBroadcastTestCase() argument
221 if (node == src_node && local_rank == src_rank) { in MakeBroadcastTestCase()
364 const int src_node, const int src_local_rank, in RunMultiNodeBroadcastTest() argument
367 const int src_global_rank = src_node * num_ranks_per_node + src_local_rank; in RunMultiNodeBroadcastTest()
373 num_nodes, num_ranks_per_node, TensorShape({5, 6}), src_node, in RunMultiNodeBroadcastTest()
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/utils/
H A Dgrappler_test.cc198 const NodeDef* src_node = node_map.GetNode(src); in IsNodesDirectlyConnected() local
200 EXPECT_TRUE(src_node != nullptr) << src << " node not found"; in IsNodesDirectlyConnected()
202 return src_node && dst_node && dst_node->input(position) == src_node->name(); in IsNodesDirectlyConnected()
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dgeneric_layout_optimizer_transposer.cc394 utils::MutableNodeView* src_node, in UpdateFanoutEdgesWithOp() argument
397 const auto* output_shape_attr = src_node->GetAttr(kAttrOutputShape); in UpdateFanoutEdgesWithOp()
406 " in", src_node->GetName()), in UpdateFanoutEdgesWithOp()
410 src_node, kAttrOutputShape, shape_attr_copy); in UpdateFanoutEdgesWithOp()
413 const bool is_in_frame = context->frames.IsInFrame(*src_node->node()); in UpdateFanoutEdgesWithOp()
418 const auto& fanouts_src_port = src_node->GetRegularFanout(src_port); in UpdateFanoutEdgesWithOp()
427 GetFanoutNameFormat(src_node->GetName(), src_port, in UpdateFanoutEdgesWithOp()
432 src_node, fanout.node_view())); in UpdateFanoutEdgesWithOp()
487 utils::MutableNodeView* src_node, utils::MutableNodeView* dst_node) { in UpdateEdge() argument
488 DCHECK(src_node != nullptr); in UpdateEdge()
[all …]
/aosp_15_r20/external/mesa3d/src/amd/vulkan/bvh/
H A Dencode.comp62 REF(radv_ir_box_node) src_node = INDEX(radv_ir_box_node, intermediate_internal_nodes, global_id);
63 radv_ir_box_node src = DEREF(src_node);
72 …uint32_t bvh_offset = is_root_node ? id_to_offset(RADV_BVH_ROOT_NODE) : DEREF(src_node).bvh_offset;
H A Dupdate.comp88 REF(radv_bvh_box32_node) src_node = REF(radv_bvh_box32_node)OFFSET(src_bvh, offset);
92 children[i] = DEREF(src_node).children[i];
/aosp_15_r20/external/e2fsprogs/lib/ext2fs/
H A Dblkmap64_rb.c251 struct rb_node *dest_node, *src_node, *dest_last, **n; in rb_copy_bmap() local
263 src_node = ext2fs_rb_first(&src_bp->root); in rb_copy_bmap()
264 while (src_node) { in rb_copy_bmap()
265 src_ext = node_to_extent(src_node); in rb_copy_bmap()
285 src_node = ext2fs_rb_next(src_node); in rb_copy_bmap()
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/tensorflow_graph_matching/
H A Dresolve_cluster.cc116 for (const std::unique_ptr<tensorflow::NodeDef>& src_node : in MaybeResolveClusters() local
119 AddNodeToGraph(*src_node, cluster_names, pruned_graph.get()); in MaybeResolveClusters()
/aosp_15_r20/external/openthread/tests/toranj/ncp/
H A Dtest-023-multicast-traffic.py64 src_node, argument
77 sender = src_node.prepare_tx(src_addr, mcast_addr, msg_len, mcast_hops=mcast_hops)
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
H A Dconfigure_tpu_embedding_rewrite_pass.cc115 for (Node* src_node : input_dependencies) { in AddExecutePartitionerNode()
116 graph->AddControlEdge(src_node, *partitioner_node); in AddExecutePartitionerNode()
H A Ddistributed_tpu_configuration_rewrite_pass.cc210 for (Node* src_node : input_dependencies) { in AddHostDisconnectNode()
211 graph->AddControlEdge(src_node, host_disconnect_node); in AddHostDisconnectNode()
/aosp_15_r20/external/coreboot/src/commonlib/
H A Ddevice_tree.c1892 struct device_tree_node *src_node; in dt_copy_subtree() local
1893 list_for_each(src_node, src->children, list_node) { in dt_copy_subtree()
1896 if (!strcmp(node->name, src_node->name)) { in dt_copy_subtree()
1903 *dst_node = *src_node; in dt_copy_subtree()
1906 dt_copy_subtree(dst_node, src_node, upd); in dt_copy_subtree()

123