Home
last modified time | relevance | path

Searched defs:new_node_inputs (Results 1 – 25 of 32) sorted by relevance

12

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ge/
Dlamb_fission.cc66 …std::vector<AnfNodePtr> new_node_inputs = {NewValueNode(std::make_shared<Primitive>(op_name)), nod… in CreateNodeOfBinaryOp() local
76 …std::vector<AnfNodePtr> new_node_inputs = {NewValueNode(std::make_shared<Primitive>(prim::kPrimUpd… in CreateUpdateStateNode() local
108 std::vector<AnfNodePtr> new_node_inputs = { in CreateLambApplyOptimizerAssignNode() local
228 std::vector<AnfNodePtr> new_node_inputs = { in CreateLambApplyWeightAssignNode() local
273 std::vector<AnfNodePtr> new_node_inputs = {NewValueNode(prim), global_step_node1}; in Process() local
Dadd_noop_to_es_grad.cc101 …std::vector<AnfNodePtr> new_node_inputs = {NewValueNode(std::make_shared<Primitive>(prim::kPrimUpd… in InsertNoOpForOutput() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fission/
Dlamb_fission.cc66 …std::vector<AnfNodePtr> new_node_inputs = {NewValueNode(std::make_shared<Primitive>(op_name)), nod… in CreateNodeOfBinaryOp() local
76 …std::vector<AnfNodePtr> new_node_inputs = {NewValueNode(std::make_shared<Primitive>(prim::kPrimUpd… in CreateUpdateStateNode() local
100 std::vector<AnfNodePtr> new_node_inputs = { in CreateLambApplyOptimizerAssignNode() local
175 std::vector<AnfNodePtr> new_node_inputs = { in CreateLambApplyWeightAssignNode() local
Dscale_grad_fission.cc36 …std::vector<AnfNodePtr> new_node_inputs = {NewValueNode(std::make_shared<Primitive>(op_name)), nod… in CreateNodeOfBinaryOp() local
41 …std::vector<AnfNodePtr> new_node_inputs = {NewValueNode(std::make_shared<Primitive>(kCastOpName)),… in CreateCastNode() local
Dadam_weight_decay_fission.cc45 …std::vector<AnfNodePtr> new_node_inputs = {NewValueNode(std::make_shared<Primitive>(op_name)), nod… in CreateNodeOfBinaryOp() local
50 …std::vector<AnfNodePtr> new_node_inputs = {NewValueNode(std::make_shared<Primitive>(op_name)), nod… in CreateNodeOfUnaryOp() local
Dmaximum_grad_fission.cc41 std::vector<AnfNodePtr> new_node_inputs = {NewValueNode(std::make_shared<Primitive>(op_name))}; in CreateCNode() local
Dascend_clip_by_norm_fission.cc75 std::vector<AnfNodePtr> new_node_inputs = {NewValueNode(std::make_shared<Primitive>(op_name))}; in CreateCNodeBase() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/mindir/
Dadam_weight_decay_unify_mindir.cc71 …std::vector<AnfNodePtr> new_node_inputs = {NewValueNode(std::make_shared<Primitive>(kSubOpName)), … in CreateSubCNode() local
79 …std::vector<AnfNodePtr> new_node_inputs = {NewValueNode(std::make_shared<Primitive>(kAssignOpName)… in CreateAssignCNode() local
140 std::vector<AnfNodePtr> new_node_inputs = {NewValueNode(prim)}; in Process() local
Dclip_by_norm_fission.cc127 std::vector<AnfNodePtr> new_node_inputs = {NewValueNode(std::make_shared<Primitive>(op_name))}; in CreateCNodeBase() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/graph/
Dargs_to_attr_pass.cc51 std::vector<AnfNodePtr> new_node_inputs; in Run() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fusion/
Dlamb_next_right_rule.cc28 std::vector<AnfNodePtr> new_node_inputs; in CreateLambNextRightNode() local
Dmomentum_lossscale_fusion.cc86 std::vector<AnfNodePtr> new_node_inputs{NewValueNode(new_prim), in Process() local
Dadam_apply_one_fusion.cc219 std::vector<AnfNodePtr> new_node_inputs = {NewValueNode(prim)}; in CreateAdamApplyOneNode() local
Dlamb_next_mv_with_decay_rule.cc67 std::vector<AnfNodePtr> new_node_inputs = {NewValueNode(prim)}; in CreateLambNextMVWithDecayNode() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fusion/
Dprelu_fusion.cc83 std::vector<AnfNodePtr> new_node_inputs = {value_node}; in Process() local
Dhard_swish_fusion.cc82 std::vector<AnfNodePtr> new_node_inputs = {value_node}; in Process() local
Dnorm_fusion.cc137 std::vector<AnfNodePtr> new_node_inputs = {value_node}; in CreateNormNode() local
467 std::vector<AnfNodePtr> new_node_inputs = {value_node}; in CreateActivationNode() local
Dgroupnorm_fusion.cc248 std::vector<AnfNodePtr> new_node_inputs = {value_node}; in CreateGroupNormNode() local
Dtf_bidirection_gru_fusion.cc583 std::vector<AnfNodePtr> new_node_inputs = {value_node, fw_init_state, bw_init_state}; in GetStackedHiddenState() local
685 std::vector<AnfNodePtr> new_node_inputs = {value_node, input, gate_weight, recu_weight, in CreateBiDirectionGruNode() local
708 std::vector<AnfNodePtr> new_node_inputs = {split_value_node, gru_output}; in GetPostProcessNode() local
Dtf_lstm_cell_fusion.cc441 …std::vector<AnfNodePtr> new_node_inputs = {value_node, input_tensor_node, i_weight, c_weight, bias… in CreateLSTMNode() local
Ddecoder_layer_fusion.cc485 std::vector<AnfNodePtr> new_node_inputs = {value_node, input, gamma1}; in CreateMaskedDecoderLayerFusionNode() local
Dtflite_rel_pos_multi_head_attention_fusion.cc275 std::vector<AnfNodePtr> new_node_inputs = {value_node, in CreateRelPosMultiHeadAttentionNode() local
Dmulti_head_attention_fusion.cc975 std::vector<AnfNodePtr> new_node_inputs = {}; in GetNewNodeInputs() local
1062 std::vector<AnfNodePtr> new_node_inputs = in CreateMaskedMultiHeadAttentionNode() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/optimizer/
Dclip_by_norm_fission.cc76 std::vector<AnfNodePtr> new_node_inputs = {NewValueNode(std::make_shared<Primitive>(op_name))}; in CreateCNodeBase() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/optimizer/
Dhelper.cc357 std::vector<AnfNodePtr> new_node_inputs = {NewValueNode(prim), node}; in CreateTensorMoveOp() local
720 AnfNodePtr CreateNodeBase(const FuncGraphPtr &graph, const std::vector<AnfNodePtr> &new_node_inputs, in CreateNodeBase()

12