Home
last modified time | relevance | path

Searched refs:kIndex6 (Results 1 – 9 of 9) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/
Dlamb_update_with_lr_v2.cc31 VectorRef select0({prim::kPrimSelect, greater1, real_div0, input_varptr_[kIndex6]}); in DefinePattern()
32 VectorRef select1({prim::kPrimSelect, greater0, select0, input_varptr_[kIndex6]}); in DefinePattern()
Dlamb_next_mv_with_decay_rule.cc149 VectorRef mul4 = VectorRef({mul4_var_, constant_mul_input_vars_[kIndex4], input_vars_[kIndex6]}); in DefinePattern()
190 VectorRef mul4 = VectorRef({mul4_var_, constant_mul_input_vars_[kIndex4], input_vars_[kIndex6]}); in DefinePattern()
231 VectorRef mul4 = VectorRef({mul4_var_, input_vars_[kIndex6], constant_mul_input_vars_[kIndex4]}); in DefinePattern()
273 VectorRef mul4 = VectorRef({mul4_var_, constant_mul_input_vars_[kIndex4], input_vars_[kIndex6]}); in DefinePattern()
Dmomentum_lossscale_fusion.cc79 …auto depend = func_graph->NewCNode({depend_prim, cnode->input(kIndex5), cnode->input(kIndex6)}); … in Process()
Dbn_reduce_grad_conv2d_backprop_filter_fusion.cc59 c[kIndex6] == out_shape[kIndex2] && c[kIndex7] == out_shape[kIndex3]); in CheckSupported()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/
Dtbe_adapter.h82 inputs_json->push_back(inputs_list[kIndex6]); in InputOrderPass()
Dtbe_adapter.cc129 inputs_json->push_back(inputs_list[kIndex6]); in InputOrderPass()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/
Ddynamic_rnn_grad_fission_v2.cc121 …auto origin_input5_shape = AnfAlgo::GetOutputInferShape(dynamic_rnn_grad_cnode->input(kIndex6), 0); in AddLSTMInputGradNode()
196 … AnfAlgo::GetOutputInferShape(dynamic_rnn_grad_cnode->input(kIndex6), 0)[0], in AddLSTMInputGradNode()
197 … AnfAlgo::GetOutputInferShape(dynamic_rnn_grad_cnode->input(kIndex6), 0)[1]}; in AddLSTMInputGradNode()
Ddynamic_gru_v2_grad_fission.cc43 {"y", kIndex4}, {"init_h", kIndex5}, {"h", kIndex6},
53 {"update", kIndex5}, {"reset", kIndex6}, {"new", kIndex7}, {"hidden_new", kIndex8}};
/third_party/mindspore/mindspore/ccsrc/utils/
Dutils.h546 kIndex6, enumerator