| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/ |
| D | adam_apply_one_fusion.cc | 25 VectorRef mul2 = VectorRef({prim::kPrimMul, mul_x_input_vars_[kIndex2], input_vars_[kIndex1]}); in DefinePattern()
29 VectorRef mul1 = VectorRef({prim::kPrimMul, mul_x_input_vars_[kIndex1], input_vars_[kIndex0]}); in DefinePattern()
40 VectorRef mul2 = VectorRef({prim::kPrimMul, mul_x_input_vars_[kIndex2], input_vars_[kIndex1]}); in DefinePattern()
44 VectorRef mul1 = VectorRef({prim::kPrimMul, mul_x_input_vars_[kIndex1], input_vars_[kIndex0]}); in DefinePattern()
55 VectorRef mul2 = VectorRef({prim::kPrimMul, mul_x_input_vars_[kIndex2], input_vars_[kIndex1]}); in DefinePattern()
59 VectorRef mul1 = VectorRef({prim::kPrimMul, mul_x_input_vars_[kIndex1], input_vars_[kIndex0]}); in DefinePattern()
70 VectorRef mul2 = VectorRef({prim::kPrimMul, mul_x_input_vars_[kIndex2], input_vars_[kIndex1]}); in DefinePattern()
74 VectorRef mul1 = VectorRef({prim::kPrimMul, mul_x_input_vars_[kIndex1], input_vars_[kIndex0]}); in DefinePattern()
85 VectorRef mul2 = VectorRef({prim::kPrimMul, mul_x_input_vars_[kIndex2], input_vars_[kIndex1]}); in DefinePattern()
89 VectorRef mul1 = VectorRef({prim::kPrimMul, mul_x_input_vars_[kIndex1], input_vars_[kIndex0]}); in DefinePattern()
[all …]
|
| D | lamb_next_mv_with_decay_v1_rule.cc | 34 auto real_div2_anf = add3->input(kIndex1); in GetSharedNodes()
44 auto add2_anf = sqrt0->input(kIndex1); in GetSharedNodes()
48 …return std::make_tuple(add3->input(kIndex2), real_div2->input(kIndex1), add2->input(kIndex1), add2… in GetSharedNodes()
60 auto real_div4_anf = add5->input(kIndex1); in MatchAdd5Pattern()
77 auto sqrt1_anf = add4->input(kIndex1); in MatchAdd5Pattern()
86 …turn add5->input(kIndex2) == mul4 && real_div4->input(kIndex1) == real_div0 && sqrt1->input(kIndex… in MatchAdd5Pattern()
192 (void)manager->Replace(add0, fusion_node_outputs[kIndex1]); in Process()
|
| D | bn_reduce_grad_conv2d_backprop_filter_fusion.cc | 57 …return (c[kIndex0] == x_shape[kIndex1] && c[kIndex1] == y_shape[kIndex1] && c[kIndex2] == x_shape[… in CheckSupported()
82 …auto bnreduce_grad = CheckAnfNodeIfCNodeAndInputSize(conv_back_filter->input(kIndex1), kBNTraining… in Process()
|
| D | lamb_next_mv_with_decay_rule.cc | 50 (void)manager->Replace(add0, new_node_outputs[kIndex1]); in GetLambNextMVWithDecayOutput()
138 …VectorRef mul2 = VectorRef({prim::kPrimMul, input_vars_[kIndex1], constant_mul_input_vars_[kIndex2… in DefinePattern()
145 …torRef mul1 = VectorRef({prim::kPrimMul, input_vars_[kIndex3], constant_mul_input_vars_[kIndex1]}); in DefinePattern()
179 …torRef mul2 = VectorRef({prim::kPrimMul, constant_mul_input_vars_[kIndex2], input_vars_[kIndex1]}); in DefinePattern()
186 …VectorRef mul1 = VectorRef({prim::kPrimMul, constant_mul_input_vars_[kIndex1], input_vars_[kIndex3… in DefinePattern()
220 …VectorRef mul2 = VectorRef({prim::kPrimMul, input_vars_[kIndex1], constant_mul_input_vars_[kIndex2… in DefinePattern()
227 …torRef mul1 = VectorRef({prim::kPrimMul, input_vars_[kIndex3], constant_mul_input_vars_[kIndex1]}); in DefinePattern()
262 …torRef mul2 = VectorRef({prim::kPrimMul, constant_mul_input_vars_[kIndex2], input_vars_[kIndex1]}); in DefinePattern()
269 …VectorRef mul1 = VectorRef({prim::kPrimMul, constant_mul_input_vars_[kIndex1], input_vars_[kIndex3… in DefinePattern()
|
| D | lamb_update_with_lr_v2.cc | 29 VectorRef greater1({prim_greater, input_varptr_[kIndex1], input_varptr_[kIndex5]}); in DefinePattern()
30 VectorRef real_div0({prim_deal_div, input_varptr_[kIndex0], input_varptr_[kIndex1]}); in DefinePattern()
|
| D | parameter_and_transop_fusion.cc | 109 AnfAlgo::GetCNodeName(trans_road[kIndex1]) == prim::kPrimCast->name() && in Run()
117 auto cast = trans_road[kIndex1]; in Run()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/buffer_fusion/ |
| D | segment_eltwise_fusion_pass.cc | 30 auto eltwise_input = cnode->input(kIndex1); in MatchSegmentEltwise()
35 eltwise_input = input_cnode->input(kIndex1); in MatchSegmentEltwise()
50 auto previous_eltwise_input = previous_input_cnode->input(kIndex1); in MatchSegmentEltwise()
56 previous_eltwise_input = previous_node->input(kIndex1); in MatchSegmentEltwise()
|
| D | reduce_eltwise_fusion_pass.cc | 32 auto eltwise_input = cnode->input(kIndex1); in MatchReduceEltwise()
37 eltwise_input = input_cnode->input(kIndex1); in MatchReduceEltwise()
53 auto previous_eltwise_input = previous_input_cnode->input(kIndex1); in MatchReduceEltwise()
59 previous_eltwise_input = previous_node->input(kIndex1); in MatchReduceEltwise()
|
| D | multi_output_fusion_pass.cc | 31 auto eltwise_input = cnode->input(kIndex1); in MatchMultiOutputEltwise()
37 eltwise_input = input_cnode->input(kIndex1); in MatchMultiOutputEltwise()
48 eltwise_input = input_cnode->input(kIndex1); in MatchMultiOutputEltwise()
|
| D | depthwiseconv_eltwise_fusion_pass.cc | 33 auto depthwise_conv = cnode->input(kIndex1); in MatchDepthwiseConvRelu()
42 auto relu = cnode->input(kIndex1); in MatchDepthwiseConvRelu()
65 auto eltwise_input = cnode->input(kIndex1); in MatchSingleFusionPattern()
|
| D | stridedread_conv_stridedwrite_fusion_pass.cc | 31 auto write_input = cnode->input(kIndex1); in MatchStridedReadConvStridedWrite()
36 write_input = input_cnode->input(kIndex1); in MatchStridedReadConvStridedWrite()
50 auto conv_input = conv_cnode->input(kIndex1); in MatchStridedReadConvStridedWrite()
|
| D | eltwise_fusion_pass.cc | 30 auto eltwise_input = cnode->input(kIndex1); in MatchEltwise()
39 eltwise_input = input_cnode->input(kIndex1); in MatchEltwise()
|
| D | conv_single_in_fusion_pass.cc | 34 auto eltwise_input = cnode->input(kIndex1); in MatchConvSingleInEltwise()
39 eltwise_input = input_cnode->input(kIndex1); in MatchConvSingleInEltwise()
|
| D | conv_double_in_fusion_pass.cc | 30 auto eltwise_input = cnode->input(kIndex1); in MatchConvDoubleInEltwise()
39 auto double_in_eltwise_input = input_cnode->input(kIndex1); in MatchConvDoubleInEltwise()
|
| D | bnupdate_eltwise_eltwise_fusion_pass.cc | 39 auto tuple_getitem = add->input(kIndex1); in MatchBnupdateAddRelu()
72 auto eltwise_input = cnode->input(kIndex1); in MatchSingleFusionPattern()
|
| D | conv2dbackprop_eltwise_eltwise_fusion_pass.cc | 30 auto eltwise_input = cnode->input(kIndex1); in MatchConv2DBackpropInputEltwiseEltwise()
52 auto double_in_eltwise_input_1 = input_cnode->input(kIndex1); in MatchConv2DBackpropInputEltwiseEltwise()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/mindir/ |
| D | maxpool_to_maxpool_with_argmax.cc | 55 maxpool->input(kIndex1)}; in CreateMaxPoolWithArgmax()
80 …ValueNode(std::make_shared<Primitive>(kMaxPoolGradWithArgmaxOpName)), maxpool_grad->input(kIndex1), in CreateMaxPoolGradWithArgmax()
81 maxpool_grad->input(kIndex3), maxpool_argmax_outputs[kIndex1]}; in CreateMaxPoolGradWithArgmax()
102 strides[kIndex1] = strides[kIndex2]; in SetNodeAttrs()
106 ksize[kIndex1] = ksize[kIndex2]; in SetNodeAttrs()
|
| D | dropout_unify_mindir.cc | 215 auto dropout_node = getitem1_cnode->input(kIndex1); in Process()
223 auto dropout_input = dropout_cnode->input(kIndex1); in Process()
292 auto dropout_node = tuple_cnode->input(kIndex1); in Process()
299 auto dropout_input = dropout_cnode->input(kIndex1); in Process()
341 auto dropout_input = dropout_node->input(kIndex1); in Process()
396 auto grad_input = dropout_grad_cnode->input(kIndex1); in Process()
|
| D | fake_learned_scale_quant_grad_unify_mindir.cc | 42 …lsq_perlayer_grad_inputs[kIndex1], lsq_perlayer_grad_inputs[kIndex2], lsq_perlayer_grad_inputs[kIn… in CreateOutputsOfLSQPerLayerGradD()
76 lsq_perlayer_grad_d_outputs[kIndex1]}; in CreateOutputsOfLSQPerLayerReduceGrad()
134 lsq_perchannel_grad_d_outputs[kIndex1]}; in CreateOutputsOfLSQPerChannelReduceGrad()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/ |
| D | tbe_adapter.h | 70 inputs_json->push_back(inputs_list[kIndex1]); in InputOrderPass()
78 inputs_json->push_back(inputs_list[kIndex1]); in InputOrderPass()
87 inputs_json->push_back(inputs_list[kIndex1]); in InputOrderPass()
106 inputs_json->push_back(inputs_list[kIndex1]); in DynamicInputAdjusted()
112 inputs_json->push_back(inputs_list[kIndex1]); in DynamicInputAdjusted()
|
| D | tbe_adapter.cc | 89 inputs_json->push_back(inputs_list[kIndex1]); in DynamicInputAdjusted()
95 inputs_json->push_back(inputs_list[kIndex1]); in DynamicInputAdjusted()
117 inputs_json->push_back(inputs_list[kIndex1]); in InputOrderPass()
125 inputs_json->push_back(inputs_list[kIndex1]); in InputOrderPass()
134 inputs_json->push_back(inputs_list[kIndex1]); in InputOrderPass()
180 (void)reorder_data_layer->emplace_back(data_layer[kIndex1]); in FusionDataOrderPass()
185 (void)reorder_data_layer->emplace_back(data_layer[kIndex1]); in FusionDataOrderPass()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/ |
| D | batch_norm_grad_split.cc | 38 NewValueNode(std::make_shared<Primitive>(kBNTrainingUpdateGradOpName)), bn_grad_inputs[kIndex1], in CreateOutputsOfUpdateGrad()
67 bn_grad_inputs[kIndex1], in CreateOutputsOfReduceGrad()
70 bn_update_grad_outputs[kIndex1], in CreateOutputsOfReduceGrad()
|
| D | bn_grad_split.cc | 39 NewValueNode(std::make_shared<Primitive>(kBNTrainingUpdateGradOpName)), bn_grad_inputs[kIndex1], in CreateOutputsOfUpdateGrad()
66 bn_grad_inputs[kIndex1], in CreateOutputsOfReduceGrad()
69 bn_update_grad_outputs[kIndex1], in CreateOutputsOfReduceGrad()
|
| D | batch_norm_bert_fission.cc | 72 NewValueNode(std::make_shared<Primitive>(kBNTrainingReduceOpName)), bn_cnode->input(kIndex1)}; in CreateBNTrainingReduce()
104 bn_cnode->input(kIndex1), in CreateBNTrainingUpdateV2()
106 bn_training_reduce_outputs[kIndex1], in CreateBNTrainingUpdateV2()
|
| D | single_batch_norm_fission.cc | 69 bn_cnode->input(kIndex1), in CreateBNTrainingUpdateV3()
71 bn_training_reduce_outputs[kIndex1], in CreateBNTrainingUpdateV3()
|