Home
last modified time | relevance | path

Searched refs:CreateMultipleOutputsOfAnfNode (Results 1 – 25 of 36) sorted by relevance

12

/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/
Ddynamic_gru_v2_grad_fission.cc68 CreateMultipleOutputsOfAnfNode(func_graph, dynamic_gru_v2_grad_cnode, kDynamicGRUV2GradOutputNum, in CreateGRUV2HiddenGradCellNode()
75 CreateMultipleOutputsOfAnfNode(func_graph, last_gru_hidden_grad_node->cast<CNodePtr>(), in CreateGRUV2HiddenGradCellNode()
133 CreateMultipleOutputsOfAnfNode(func_graph, gru_hidden_grad_cnode, kGRUV2HiddenGradCellOutputNum, in AddTLoopNode()
173 … CreateMultipleOutputsOfAnfNode(func_graph, gru_hidden_grad_node_i, kGRUV2HiddenGradCellOutputNum, in AddTConcatNode()
264 CreateMultipleOutputsOfAnfNode(func_graph, splitv, kSplitVOutputNum, &splitv_outputs); in AddHConcatNode()
295 …CreateMultipleOutputsOfAnfNode(func_graph, dgate_h, kGRUV2HiddenGradCellOutputNum, &dgate_h_output… in AddDwhMatmulNode()
315 …CreateMultipleOutputsOfAnfNode(func_graph, dgate_h, kGRUV2HiddenGradCellOutputNum, &dgate_h_output… in CreateDgateHSplitVDNode()
339 CreateMultipleOutputsOfAnfNode(func_graph, split, kSplitVOutputNum, &split_outputs); in CreateDgateXConcatDNode()
344 … CreateMultipleOutputsOfAnfNode(func_graph, dnt_x, kGRUV2HiddenGradCellOutputNum, &dnt_x_outputs); in CreateDgateXConcatDNode()
470 …CreateMultipleOutputsOfAnfNode(func_graph, dynamic_gru_v2_grad_cnode, kDynamicGRUV2GradOutputNum, … in Process()
[all …]
Ddynamic_rnn_grad_fission_v2.cc137 CreateMultipleOutputsOfAnfNode(func_graph, lstm_split_c, num_split_x, &lstm_split_c_outputs); in AddLSTMInputGradNode()
143 CreateMultipleOutputsOfAnfNode(func_graph, lstm_split_dy, num_split_x, &lstm_split_dy_outputs); in AddLSTMInputGradNode()
149 CreateMultipleOutputsOfAnfNode(func_graph, lstm_split_i, num_split_x, &lstm_split_i_outputs); in AddLSTMInputGradNode()
155 CreateMultipleOutputsOfAnfNode(func_graph, lstm_split_j, num_split_x, &lstm_split_j_outputs); in AddLSTMInputGradNode()
161 CreateMultipleOutputsOfAnfNode(func_graph, lstm_split_f, num_split_x, &lstm_split_f_outputs); in AddLSTMInputGradNode()
167 CreateMultipleOutputsOfAnfNode(func_graph, lstm_split_o, num_split_x, &lstm_split_o_outputs); in AddLSTMInputGradNode()
173 …CreateMultipleOutputsOfAnfNode(func_graph, lstm_split_tanh, num_split_x, &lstm_split_tanh_outputs); in AddLSTMInputGradNode()
222 CreateMultipleOutputsOfAnfNode(func_graph, basic_lstm_cell_c_state_grad, kBasicCellOutputNum, in AddLSTMInputGradNode()
245 CreateMultipleOutputsOfAnfNode(func_graph, split_v, kSplitVOutputNum, &split_outputs); in AddLSTMInputGradNode()
323 CreateMultipleOutputsOfAnfNode(func_graph, splitv, kSplitVOutputNum, &splitv_outputs); in CreateHConcat()
Dlayer_norm_grad_split.cc59 CreateMultipleOutputsOfAnfNode(graph, layer_norm_x_backprop, kLayerNormXBackpropV2OutputNum, in CreateOutputsOfLayerNormXBackpropV2()
91 …CreateMultipleOutputsOfAnfNode(graph, layer_norm_beta_gamma_backprop, kLayerNormBetaGammaBackpropO… in CreateOutputsOfLayerNormBetaGammaBackpropV2()
Dbatch_norm_bert_fission.cc152 CreateMultipleOutputsOfAnfNode(func_graph, bn_training_reduce, kBNTrainingReduceOutputNum, in Process()
157 CreateMultipleOutputsOfAnfNode(func_graph, bn_training_update_v2, kBNTrainingUpdateV2OutputNum, in Process()
Dlars_v2_fission.cc47 …CreateMultipleOutputsOfAnfNode(graph, square_sum_all, kSquareSumOutputNum, square_sum_all_outputs); in CreateOutputsOfSquareSumAll()
Dsingle_batch_norm_fission.cc112 CreateMultipleOutputsOfAnfNode(func_graph, bn_training_reduce, kBNTrainingReduceOutputNum, in Process()
Dbatch_norm_grad_split.cc50 …CreateMultipleOutputsOfAnfNode(graph, bn_update_grad, kBNTrainingUpdateGradOutputNum, bn_update_gr… in CreateOutputsOfUpdateGrad()
Dbn_grad_split.cc51 …CreateMultipleOutputsOfAnfNode(graph, bn_update_grad, kBNTrainingUpdateGradOutputNum, bn_update_gr… in CreateOutputsOfUpdateGrad()
Dbatch_norm_grad_infer_fission.cc170 …CreateMultipleOutputsOfAnfNode(func_graph, bn_training_update_grad, kBNTrainingUpdateGradOutputNum, in Process()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/mindir/
Dfake_learned_scale_quant_grad_unify_mindir.cc55 CreateMultipleOutputsOfAnfNode(graph, lsq_perlayer_grad_d, kFakeLearnedScaleQuantGradDOutputNum, in CreateOutputsOfLSQPerLayerGradD()
113 CreateMultipleOutputsOfAnfNode(graph, lsq_perchannel_grad_d, kFakeLearnedScaleQuantGradDOutputNum, in CreateOutputsOfLSQPerChannelGradD()
Dsparse_softmax_cross_entropy_with_logits_unify_mindir.cc414 …CreateMultipleOutputsOfAnfNode(graph, softmax_node, kSoftmaxCrossEntropyWithLogitsOutputNum, &soft… in Process()
451 …CreateMultipleOutputsOfAnfNode(graph, softmax_node, kSoftmaxCrossEntropyWithLogitsOutputNum, &soft… in Process()
503 …CreateMultipleOutputsOfAnfNode(graph, softmax_node, kSoftmaxCrossEntropyWithLogitsOutputNum, &soft… in Process()
528 …CreateMultipleOutputsOfAnfNode(graph, softmax_node, kSoftmaxCrossEntropyWithLogitsOutputNum, &soft… in Process()
568 …CreateMultipleOutputsOfAnfNode(graph, softmax_node, kSoftmaxCrossEntropyWithLogitsOutputNum, &soft… in Process()
Dall_to_all_unify_mindir.cc100 CreateMultipleOutputsOfAnfNode(graph, split, static_cast<size_t>(split_count), &split_outputs); in CreateAllToAllvNode()
133 …CreateMultipleOutputsOfAnfNode(graph, all_to_all_v, static_cast<size_t>(split_count), &all_to_all_… in CreateConcatNode()
Dmaxpool_to_maxpool_with_argmax.cc139 …CreateMultipleOutputsOfAnfNode(graph, maxpool_argmax, kMaxPoolWithArgmaxOutputNum, &maxpool_argmax… in Process()
/third_party/mindspore/mindspore/lite/tools/optimizer/parallel/
Doperator_info.cc106 int OperatorInfo::CreateMultipleOutputsOfAnfNode(const AnfNodePtr &node, size_t output_num, in CreateMultipleOutputsOfAnfNode() function in mindspore::opt::OperatorInfo
160 (void)CreateMultipleOutputsOfAnfNode(concat_cnode, 1, &outputs); in CreateConcateNode()
Doperator_info.h65 …int CreateMultipleOutputsOfAnfNode(const AnfNodePtr &node, size_t output_num, std::vector<AnfNodeP…
Dconv2d_info.cc199 if (CreateMultipleOutputsOfAnfNode(split_cnode, split_num, split_outputs) != RET_OK) { in CreateOutputsOfSplit()
361 (void)CreateMultipleOutputsOfAnfNode(conv_cnode, 1, &tmp_outputs); in ConstructOutputCNodes()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fusion/
Dlamb_next_right_rule.cc85 CreateMultipleOutputsOfAnfNode(func_graph, new_node, kLambNextRightOutputNum, &new_node_outputs); in Process()
Dadd_input_to_output.cc109 CreateMultipleOutputsOfAnfNode(func_graph, cnode, new_abstract_tuple->size(), &new_outputs); in Process()
Dbn_reduce_grad_conv2d_backprop_filter_fusion.cc110 …CreateMultipleOutputsOfAnfNode(graph, fused_dbn_dw, kFusedDbnDwDropoutOutputNum, &fused_dbn_dw_out… in Process()
Dderelu_fusion.cc120 CreateMultipleOutputsOfAnfNode(graph, relu_v2, kReluV2OutputNum, &relu_v2_node_outputs); in Process()
Dsquare_sum_fusion.cc126 … CreateMultipleOutputsOfAnfNode(graph, square_sumv2, kSquareSumv2OutputNum, &square_sumv2_outputs); in Process()
Dfused_batch_norm_fusion.cc200 CreateMultipleOutputsOfAnfNode(func_graph, bn_training_reduce, kBNTrainingReduceOutputNum, in Process()
208 CreateMultipleOutputsOfAnfNode(func_graph, bn_training_update, kBNTrainingUpdateOutputNum, in Process()
Dconfusion_mul_grad_fusion.cc151 …CreateMultipleOutputsOfAnfNode(graph, fusion_node, kConfusionMulGradOutputNum, &fusion_node_output… in Process()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/enhancer/
Dsplit_inputs_for_reduce_scatter.cc49 …CreateMultipleOutputsOfAnfNode(func_graph, split, AnfAlgo::GetOutputTensorNum(split), &new_outputs… in InsertSplitForInput()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/
Drelu_v2_pass.cc143 CreateMultipleOutputsOfAnfNode(graph, relu_v2, kReluV2OutputNum, &relu_v2_node_outputs); in Process()

12