Home
last modified time | relevance | path

Searched refs:GetInputNode (Results 1 – 25 of 65) sorted by relevance

123

/third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/
Dbatch_norm_relu_grad_fusion.cc63 auto relu_grad = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(node), 0); in Process()
72 auto dy = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(relu_grad), 0); in Process()
74 auto y = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(relu_grad), 1); in Process()
76 auto x = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(node), 1); in Process()
78 auto scale = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(node), 2); in Process()
80 auto save_mean = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(node), 3); in Process()
82 auto save_var = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(node), 4); in Process()
84 auto reserve = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(node), 5); in Process()
86 auto batch_norm = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(save_mean), 0); in Process()
88 auto bias = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(batch_norm), 2); in Process()
Dbatch_norm_add_relu_fusion.cc43 auto tensor_add = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(node), 0); in Process()
45 auto tuple_get_item = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(tensor_add), 0); in Process()
47 auto batch_norm = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(tuple_get_item), 0); in Process()
65 auto x = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(batch_norm), 0); in Process()
66 auto scale = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(batch_norm), 1); in Process()
67 auto bias = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(batch_norm), 2); in Process()
68 auto mean = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(batch_norm), 3); in Process()
69 auto var = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(batch_norm), 4); in Process()
70 auto z = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(tensor_add), 1); in Process()
Dpost_batch_norm_add_relu_fusion.cc43 auto tensor_add = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(node), 0); in Process()
45 auto tuple_get_item = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(tensor_add), 1); in Process()
47 auto batch_norm = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(tuple_get_item), 0); in Process()
65 auto x = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(batch_norm), 0); in Process()
66 auto scale = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(batch_norm), 1); in Process()
67 auto bias = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(batch_norm), 2); in Process()
68 auto mean = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(batch_norm), 3); in Process()
69 auto var = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(batch_norm), 4); in Process()
70 auto z = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(tensor_add), 0); in Process()
Dbatch_norm_add_relu_grad_fusion.cc129 auto relu_grad = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(node), 0); in PatternCheck()
137 auto tuple_getitem = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(node), 5); in PatternCheck()
142 auto forward_node = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(tuple_getitem), 0); in PatternCheck()
167 auto relu_grad = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(node), 0); in Process()
169 auto dy = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(relu_grad), 0); in Process()
171 auto y = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(relu_grad), 1); in Process()
173 auto x = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(node), 1); in Process()
175 auto scale = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(node), 2); in Process()
177 auto save_mean = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(node), 3); in Process()
179 auto save_var = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(node), 4); in Process()
[all …]
Dbatch_norm_relu_fusion.cc42 auto tuple_get_item = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(node), 0); in Process()
44 auto batch_norm = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(tuple_get_item), 0); in Process()
62 auto x = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(batch_norm), 0); in Process()
63 auto scale = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(batch_norm), 1); in Process()
64 auto bias = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(batch_norm), 2); in Process()
65 auto mean = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(batch_norm), 3); in Process()
66 auto var = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(batch_norm), 4); in Process()
Dremove_format_transform_pair.cc41 auto input_node = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(node), 0); in Process()
57 auto transpose1_input_node = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(input_node), 0); in Process()
Dreplace_addn_fusion.cc32 auto A = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(node), 0); in Process()
33 auto B = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(node), 1); in Process()
Dreplace_momentum_cast_fusion.cc35 auto grad_cast = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(node), kGradIndex); in Process()
42 auto grad = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(grad_cast), 0); in Process()
Dcombine_cast_fusion.cc39 auto input0 = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(cast_node), 0); in GetDealList()
41 … (AnfAlgo::GetInputNode(utils::cast<CNodePtr>(input0), 0))->isa<Parameter>())) { in GetDealList()
96 inputs.push_back(AnfAlgo::GetInputNode(cnode, 0)); in Run()
Dcudnn_inplace_fusion.cc80 const AnfNodePtr &input_node = AnfAlgo::GetInputNode(current_node, input_index); in ExistRoute()
260 *skip_node = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(node), aggregate_iter->second); in PatternMatch()
271 auto inplace_node = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(*skip_node), i); in PatternMatch()
282 inplace_node = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(inplace_node), 0); in PatternMatch()
Dprint_reduce_fusion.cc71 auto current_node = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(node), i); in GetOptList()
152 auto input_tensor = AnfAlgo::GetInputNode(cnode, i); in Run()
157 auto monad_node = AnfAlgo::GetInputNode(cnode, input_flag.size()); in Run()
Dremove_redundant_format_transform.cc39 auto input_node = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(node), 0); in Process()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/pass/
Dadjust_depend_for_parallel_optimizer_recompute_all_gather.cc105 auto depend_node = AnfAlgo::GetInputNode(cnode, 0); in AdjustAllgatherDepend()
116 … allgather_next_node, AnfAlgo::GetInputNode(depend_cnode, 1)}; in AdjustAllgatherDepend()
119 manager->SetEdge(node, 1, AnfAlgo::GetInputNode(depend_cnode, 0)); in AdjustAllgatherDepend()
124 … IsPrimitiveCNode(AnfAlgo::GetInputNode(depend_node->cast<CNodePtr>(), 0), prim::kPrimDepend)) { in AdjustAllgatherDepend()
126 auto cast_depend_node = AnfAlgo::GetInputNode(cast_cnode, 0); in AdjustAllgatherDepend()
136 … allgather_next_node, AnfAlgo::GetInputNode(cast_depend_cnode, 1)}; in AdjustAllgatherDepend()
139 manager->SetEdge(depend_node, 1, AnfAlgo::GetInputNode(cast_depend_cnode, 0)); in AdjustAllgatherDepend()
Dconvert_tuple_input_to_dynamic_input.cc46 auto dyn_input_node = AnfAlgo::GetInputNode(make_tuple, j); in SplitTupleInputs()
71 auto input_node = AnfAlgo::GetInputNode(cnode_ptr, i); in ConvertMakeTupleInputToPlantInputs()
Doptimize_dependence.cc58 auto virtual_input_op = AnfAlgo::GetInputNode(cnode, kIsolatedDependVirtualInputIndex); in CheckIsolatedVirtualNode()
62 auto real_input_op = AnfAlgo::GetInputNode(cnode, kIsolatedDependRealInputIndex); in CheckIsolatedVirtualNode()
124 auto input = AnfAlgo::GetInputNode(cnode, index); in ReplaceMakeTuple()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/
Dremove_reshape_pair.cc41 …CheckAnfNodeIfCNodeAndInputSize(AnfAlgo::GetInputNode(out_reshape, 0), kBackendReshapeInputTensorN… in Process()
53 auto input_node = AnfAlgo::GetInputNode(in_reshape, 0); in Process()
Dparameter_and_transop_fusion.cc49 auto next_node = AnfAlgo::GetInputNode(cnode, 0); in ParamTransRoad()
106 …auto final_node = ParamTransRoad(func_graph, AnfAlgo::GetInputNode(cnode, input_index), first_flag… in Run()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/enhancer/
Dinsert_depend_for_all_gather.cc47 AnfAlgo::GetInputNode(next_cnode, 0), current_node}; in Run()
49 new_input->set_abstract(AnfAlgo::GetInputNode(next_cnode, 0)->abstract()); in Run()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/
Dreorder_ops.cc104 auto node_input = AnfAlgo::GetInputNode(new_node, 0); in SetNodeInfo()
217 auto node_input = AnfAlgo::GetInputNode(node, 0); in ReorderTypeInsensitiveCastDown()
236 …func_graph->NewCNode({NewValueNode(prim::kPrimCast), AnfAlgo::GetInputNode(type_insens_node, index… in ReorderTypeInsensitiveCastDown()
274 auto node_input = AnfAlgo::GetInputNode(node, index); in ReorderCastUpTypeInsensitive()
280 cast_input_nodes.push_back(AnfAlgo::GetInputNode(cast_node, 0)); in ReorderCastUpTypeInsensitive()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/format_type/
Dinsert_transpose_for_dynamic_gru_v2.cc52 AnfNodePtr transdata_node = AnfAlgo::GetInputNode(cnode, index); in Insert()
59 …NewTransOpNode(func_graph, AnfAlgo::GetInputNode(transdata_node->cast<CNodePtr>(), 0), kernel_sele… in Insert()
Dconvert_cast_format.cc47 …auto input_node = AnfAlgo::VisitKernelWithReturnType(AnfAlgo::GetInputNode(cnode, input_index), 0)… in Process()
124 …if (AnfAlgo::VisitKernelWithReturnType(AnfAlgo::GetInputNode(cast_out_node->cast<CNodePtr>(), inde… in CalculateFormat()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/
Dtransdata_split.cc76 …new_transdata_node = NewTransOpNode(func_graph, AnfAlgo::GetInputNode(node->cast<CNodePtr>(), 0), … in DoSplit()
86 …new_transpose_node = NewTransOpNode(func_graph, AnfAlgo::GetInputNode(node->cast<CNodePtr>(), 0), … in DoSplit()
/third_party/mindspore/mindspore/ccsrc/ps/
Dutil.cc153 … AnfAlgo::GetInputNode(node->cast<CNodePtr>(), kNodeInputWeightNameOffset)->cast<ValueNodePtr>(); in DoFusion()
158 … AnfAlgo::GetInputNode(node->cast<CNodePtr>(), kNodeInputWeightIndexOffset)->cast<ValueNodePtr>(); in DoFusion()
170 fused_node_inputs.push_back(AnfAlgo::GetInputNode(node->cast<CNodePtr>(), 0)); in DoFusion()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/trt_pass/
Dgraph_partitioner.cc140 auto input_node = AnfAlgo::GetInputNode(node->cast<CNodePtr>(), input_index); in ExistCycleAfterMerge()
164 auto input_node = AnfAlgo::GetInputNode(node->cast<CNodePtr>(), input_index); in MergeParentBranchRecursively()
193 auto input_node = AnfAlgo::GetInputNode(node->cast<CNodePtr>(), input_index); in NodeGrouping()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/cpu/
Dinsert_cast_cpu.cc78 auto cur_input = AnfAlgo::GetInputNode(cnode, input_index); in InsertCast()
111 auto cur_input = AnfAlgo::GetInputNode(utils::cast<CNodePtr>(used_node), used_node_index); in InsertCastForGraphOutput()

123