Searched refs:kIndex5 (Results 1 – 12 of 12) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/ |
| D | momentum_lossscale_fusion.cc | 79 …auto depend = func_graph->NewCNode({depend_prim, cnode->input(kIndex5), cnode->input(kIndex6)}); … in Process()
80 depend->set_abstract(cnode->input(kIndex5)->abstract()); in Process()
81 depend->set_scope(cnode->input(kIndex5)->scope()); in Process()
|
| D | lamb_update_with_lr_v2.cc | 28 VectorRef greater0({prim_greater, input_varptr_[kIndex0], input_varptr_[kIndex5]}); in DefinePattern()
29 VectorRef greater1({prim_greater, input_varptr_[kIndex1], input_varptr_[kIndex5]}); in DefinePattern()
|
| D | lamb_next_mv_with_decay_rule.cc | 147 VectorRef real_div0 = VectorRef({real_div0_var_, add0, input_vars_[kIndex5]}); in DefinePattern()
188 VectorRef real_div0 = VectorRef({real_div0_var_, add0, input_vars_[kIndex5]}); in DefinePattern()
229 VectorRef real_div0 = VectorRef({real_div0_var_, add0, input_vars_[kIndex5]}); in DefinePattern()
271 VectorRef real_div0 = VectorRef({real_div0_var_, add0, input_vars_[kIndex5]}); in DefinePattern()
|
| 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()
73 bn_grad_inputs[kIndex5]}; in CreateOutputsOfReduceGrad()
|
| D | bn_grad_split.cc | 40 bn_grad_inputs[kIndex2], bn_grad_inputs[kIndex4], bn_grad_inputs[kIndex5]}; in CreateOutputsOfUpdateGrad()
72 bn_grad_inputs[kIndex5]}; 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}};
53 {"update", kIndex5}, {"reset", kIndex6}, {"new", kIndex7}, {"hidden_new", kIndex8}};
512 auto dbx_reduce_sum = CreateDbReduceSumDNode(func_graph, dgate_x_concat, ori_inputs[kIndex5]); in Process()
517 dbh_reduce_sum = CreateDbReduceSumDNode(func_graph, dbh_outputs[kIndex1], ori_inputs[kIndex5]); in Process()
519 dbh_reduce_sum = CreateDbReduceSumDNode(func_graph, dgate_h, ori_inputs[kIndex5]); in Process()
|
| D | bn_split.cc | 88 bn_training_update_inputs.push_back(bn_cnode->input(kIndex5)); in CreateOutputsOfBNTrainingUpdate()
|
| D | dynamic_rnn_grad_fission_v2.cc | 328 auto origin_input4 = dynamic_rnn_grad_cnode->input(kIndex5); in CreateHConcat()
384 auto origin_input4 = dynamic_rnn_grad_cnode->input(kIndex5); in CreateConcatNodeT1()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/ |
| D | tbe_adapter.h | 81 inputs_json->push_back(inputs_list[kIndex5]); in InputOrderPass()
|
| D | tbe_adapter.cc | 128 inputs_json->push_back(inputs_list[kIndex5]); in InputOrderPass()
|
| /third_party/mindspore/mindspore/ccsrc/utils/ |
| D | utils.h | 545 kIndex5, enumerator
|