| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/ |
| D | lamb_next_mv_with_decay_rule.cc | 144 …VectorRef mul0 = VectorRef({prim::kPrimMul, input_vars_[kIndex4], constant_mul_input_vars_[kIndex0… in DefinePattern()
149 VectorRef mul4 = VectorRef({mul4_var_, constant_mul_input_vars_[kIndex4], input_vars_[kIndex6]}); in DefinePattern()
185 …torRef mul0 = VectorRef({prim::kPrimMul, constant_mul_input_vars_[kIndex0], input_vars_[kIndex4]}); in DefinePattern()
190 VectorRef mul4 = VectorRef({mul4_var_, constant_mul_input_vars_[kIndex4], input_vars_[kIndex6]}); in DefinePattern()
226 …VectorRef mul0 = VectorRef({prim::kPrimMul, input_vars_[kIndex4], constant_mul_input_vars_[kIndex0… in DefinePattern()
231 VectorRef mul4 = VectorRef({mul4_var_, input_vars_[kIndex6], constant_mul_input_vars_[kIndex4]}); in DefinePattern()
268 …torRef mul0 = VectorRef({prim::kPrimMul, constant_mul_input_vars_[kIndex0], input_vars_[kIndex4]}); in DefinePattern()
273 VectorRef mul4 = VectorRef({mul4_var_, constant_mul_input_vars_[kIndex4], input_vars_[kIndex6]}); in DefinePattern()
|
| D | adam_apply_one_fusion.cc | 34 …{prim::kPrimSub, input_vars_[kIndex3], VectorRef({prim::kPrimMul, input_vars_[kIndex4], true_div0}… in DefinePattern()
49 …{prim::kPrimSub, input_vars_[kIndex3], VectorRef({prim::kPrimMul, input_vars_[kIndex4], true_div0}… in DefinePattern()
64 …im::kPrimSub, input_vars_[kIndex3], VectorRef({prim::kPrimMul, true_div0, input_vars_[kIndex4]})}); in DefinePattern()
79 …im::kPrimSub, input_vars_[kIndex3], VectorRef({prim::kPrimMul, true_div0, input_vars_[kIndex4]})}); in DefinePattern()
94 …im::kPrimSub, input_vars_[kIndex3], VectorRef({prim::kPrimMul, true_div0, input_vars_[kIndex4]})}); in DefinePattern()
110 …VectorRef({sub0_var_, input_vars_[kIndex3], VectorRef({prim::kPrimMul, input_vars_[kIndex4], true_… in DefinePattern()
132 …VectorRef({sub0_var_, input_vars_[kIndex3], VectorRef({prim::kPrimMul, input_vars_[kIndex4], true_… in DefinePattern()
154 …f({sub0_var_, input_vars_[kIndex3], VectorRef({prim::kPrimMul, true_div0, input_vars_[kIndex4]})}); in DefinePattern()
176 …f({sub0_var_, input_vars_[kIndex3], VectorRef({prim::kPrimMul, true_div0, input_vars_[kIndex4]})}); in DefinePattern()
198 …f({sub0_var_, input_vars_[kIndex3], VectorRef({prim::kPrimMul, true_div0, input_vars_[kIndex4]})}); in DefinePattern()
|
| D | lamb_update_with_lr_v2.cc | 36 return VectorRef({prim::kPrimSub, input_varptr_[kIndex4], mul1}); in DefinePattern()
|
| D | momentum_lossscale_fusion.cc | 60 AnfNodePtr mul = cnode->input(kIndex4); in Process()
|
| D | bn_reduce_grad_conv2d_backprop_filter_fusion.cc | 58 …c[kIndex3] == x_shape[kIndex3] && c[kIndex4] == y_shape[kIndex2] && c[kIndex5] == y_shape[kIndex3]… in CheckSupported()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/ |
| D | batch_norm_grad_split.cc | 39 bn_grad_inputs[kIndex2], bn_grad_inputs[kIndex4], bn_grad_inputs[kIndex5]}; in CreateOutputsOfUpdateGrad()
72 bn_grad_inputs[kIndex4], in CreateOutputsOfReduceGrad()
|
| D | bn_grad_split.cc | 40 bn_grad_inputs[kIndex2], bn_grad_inputs[kIndex4], bn_grad_inputs[kIndex5]}; in CreateOutputsOfUpdateGrad()
71 bn_grad_inputs[kIndex4], in CreateOutputsOfReduceGrad()
|
| D | dynamic_gru_v2_grad_fission.cc | 43 {"y", kIndex4}, {"init_h", kIndex5}, {"h", kIndex6},
49 … {"db_hidden", kIndex3}, {"dx", kIndex4}, {"dh_prev", kIndex5}};
52 {"dh_pre_t", kIndex1}, {"h", kIndex2}, {"dy", kIndex3}, {"dh", kIndex4},
|
| D | batch_norm_bert_fission.cc | 120 bn_abstract_tuple->elements()[kIndex4]}; in CreateBNTrainingUpdateV2()
|
| D | bn_split.cc | 87 bn_training_update_inputs.push_back(bn_cnode->input(kIndex4)); in CreateOutputsOfBNTrainingUpdate()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/mindir/ |
| D | slice_grad_unify_mindir.cc | 85 sizes = GetTupleValue(slice_grad->input(kIndex4)); in Process()
|
| D | fake_learned_scale_quant_grad_unify_mindir.cc | 43 lsq_perlayer_grad_inputs[kIndex4]}; in CreateOutputsOfLSQPerLayerGradD()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/ |
| D | tbe_adapter.h | 85 inputs_json->push_back(inputs_list[kIndex4]); in InputOrderPass()
|
| D | tbe_adapter.cc | 132 inputs_json->push_back(inputs_list[kIndex4]); in InputOrderPass()
|
| /third_party/mindspore/mindspore/ccsrc/utils/ |
| D | utils.h | 544 kIndex4, enumerator
|