Home
last modified time | relevance | path

Searched refs:make_tuple_inputs (Results 1 – 25 of 35) sorted by relevance

12

/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/enhancer/
Dinsert_tensor_move_for_getnext.cc37 std::vector<AnfNodePtr> make_tuple_inputs; in InsertTensorMoveForGetNextOutputs() local
38 make_tuple_inputs.push_back(NewValueNode(prim::kPrimMakeTuple)); in InsertTensorMoveForGetNextOutputs()
50 make_tuple_inputs.push_back(new_node); in InsertTensorMoveForGetNextOutputs()
52 AnfNodePtr make_tuple = func_graph->NewCNode(make_tuple_inputs); in InsertTensorMoveForGetNextOutputs()
Dconcat_outputs_for_all_gather.cc94 …std::vector<AnfNodePtr> make_tuple_inputs{NewValueNode(std::make_shared<Primitive>(prim::kPrimMake… in InsertConcatForOutput() local
119 make_tuple_inputs.push_back(concat); in InsertConcatForOutput()
122 auto make_tuple = func_graph->NewCNode(make_tuple_inputs); in InsertConcatForOutput()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/format_type/
Dinsert_transpose_for_basiclstm_op.cc61 std::vector<AnfNodePtr> make_tuple_inputs = {NewValueNode(prim::kPrimMakeTuple)}; in Insert() local
74 make_tuple_inputs.push_back(transpose); in Insert()
76 make_tuple_inputs.push_back(tuple_getitem); in Insert()
79 new_node = func_graph->NewCNode(make_tuple_inputs); in Insert()
Dinsert_cast.cc43 std::vector<AnfNodePtr> make_tuple_inputs; in InsertCastForMultipleOutput() local
45 make_tuple_inputs.emplace_back(NewValueNode(prim::kPrimMakeTuple)); in InsertCastForMultipleOutput()
77 make_tuple_inputs.push_back(replace_node); in InsertCastForMultipleOutput()
79 AnfNodePtr make_tuple = func_graph->NewCNode(make_tuple_inputs); in InsertCastForMultipleOutput()
Ddeal_ref_and_split_unsupported_transdata.cc184 std::vector<AnfNodePtr> make_tuple_inputs; in DealRefForMultipleOutput() local
186 (void)make_tuple_inputs.emplace_back(NewValueNode(prim::kPrimMakeTuple)); in DealRefForMultipleOutput()
197 make_tuple_inputs.push_back(final_node); in DealRefForMultipleOutput()
200 CNodePtr make_tuple = func_graph->NewCNode(make_tuple_inputs); in DealRefForMultipleOutput()
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/
Dupdatestate_eliminate.cc254 const AnfNodePtrList &make_tuple_inputs) { in MakeTupleForSameNodes() argument
256 if (make_tuple_inputs.size() == kOneNodeInputSize) { in MakeTupleForSameNodes()
258 return make_tuple_inputs.at(1); in MakeTupleForSameNodes()
261 auto make_tuple = fg->NewCNode(make_tuple_inputs); in MakeTupleForSameNodes()
264 …std::transform(make_tuple_inputs.begin() + 1, make_tuple_inputs.end(), std::back_inserter(element_… in MakeTupleForSameNodes()
349 AnfNodePtrList make_tuple_inputs; in EliminateUpdateStateForLoads() local
351 make_tuple_inputs.reserve(loads.size() + 1); in EliminateUpdateStateForLoads()
352 make_tuple_inputs.emplace_back(NewValueNode(prim::kPrimMakeTuple)); in EliminateUpdateStateForLoads()
360 make_tuple_inputs.emplace_back(load); in EliminateUpdateStateForLoads()
371 if (make_tuple_inputs.size() == 1) { in EliminateUpdateStateForLoads()
[all …]
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/ir_fission/
Dbn_grad_split.cc102 …std::vector<AnfNodePtr> make_tuple_inputs = {NewValueNode(prim::kPrimMakeTuple), bn_reduce_grad_ou… in BNGradSplitForTBE() local
104 auto make_tuple = func_graph->NewCNode(make_tuple_inputs); in BNGradSplitForTBE()
133 …std::vector<AnfNodePtr> make_tuple_inputs = {NewValueNode(prim::kPrimMakeTuple), bn_reduce_grad_ou… in SyncBNGradSplitForTBE() local
135 auto make_tuple = func_graph->NewCNode(make_tuple_inputs); in SyncBNGradSplitForTBE()
Dsplit_fission.cc138 std::vector<AnfNodePtr> make_tuple_inputs = {NewValueNode(prim::kPrimMakeTuple)}; in DoFission() local
151 AddNewOutputs(func_graph, new_splitv, divisor, &make_tuple_inputs); in DoFission()
170 AddNewOutputs(func_graph, new_splitv, last_node_num_split, &make_tuple_inputs); in DoFission()
175 make_tuple_inputs.emplace_back(tuple_getitem); in DoFission()
182 AnfNodePtr make_tuple = func_graph->NewCNode(make_tuple_inputs); in DoFission()
Dbatch_norm_grad_split.cc125 …std::vector<AnfNodePtr> make_tuple_inputs = {NewValueNode(prim::kPrimMakeTuple), bn_reduce_grad_ou… in Process() local
127 auto make_tuple = func_graph->NewCNode(make_tuple_inputs); in Process()
Dlayer_norm_grad_split.cc127 …std::vector<AnfNodePtr> make_tuple_inputs = {NewValueNode(prim::kPrimMakeTuple), layer_norm_x_back… in Process() local
130 auto make_tuple = graph->NewCNode(make_tuple_inputs); in Process()
Ddynamic_rnn_grad_fission_v2.cc558 std::vector<AnfNodePtr> make_tuple_inputs = {NewValueNode(prim::kPrimMakeTuple)}; in Process() local
561 (void)make_tuple_inputs.emplace_back(dw_reduce_sum); in Process()
564 (void)make_tuple_inputs.emplace_back(dw_reshape); in Process()
570 (void)make_tuple_inputs.emplace_back(db_reduce_sum); in Process()
571 make_tuple_inputs.insert(make_tuple_inputs.end(), new_outputs.begin(), new_outputs.end()); in Process()
572 auto make_tuple = func_graph->NewCNode(make_tuple_inputs); in Process()
Dbatch_norm_grad_infer_fission.cc177 std::vector<AnfNodePtr> make_tuple_inputs = {NewValueNode(prim::kPrimMakeTuple), bn_infer_grad, in Process() local
179 auto make_tuple = func_graph->NewCNode(make_tuple_inputs); in Process()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/mindir/
Dfake_learned_scale_quant_grad_unify_mindir.cc176 …std::vector<AnfNodePtr> make_tuple_inputs = {NewValueNode(prim::kPrimMakeTuple), lsq_perlayer_grad… in Process() local
178 auto make_tuple = func_graph->NewCNode(make_tuple_inputs); in Process()
213 …std::vector<AnfNodePtr> make_tuple_inputs = {NewValueNode(prim::kPrimMakeTuple), lsq_perchannel_gr… in Process() local
215 auto make_tuple = func_graph->NewCNode(make_tuple_inputs); in Process()
Ddropout_unify_mindir.cc315 …std::vector<AnfNodePtr> make_tuple_inputs{NewValueNode(prim::kPrimMakeTuple), dropout_do_mask, dro… in Process() local
316 auto make_tuple = func_graph->NewCNode(make_tuple_inputs); in Process()
355 …std::vector<AnfNodePtr> make_tuple_inputs{NewValueNode(prim::kPrimMakeTuple), dropout_do_mask, dro… in Process() local
356 auto make_tuple = func_graph->NewCNode(make_tuple_inputs); in Process()
/third_party/mindspore/mindspore/lite/tools/converter/import/
Dmindir_control_flow_adjust.cc129 std::vector<AnfNodePtr> make_tuple_inputs = after_fg->get_inputs(); in AddAfterFuncGraph() local
137 make_tuple_inputs.insert(make_tuple_inputs.begin(), make_tuple_prim); in AddAfterFuncGraph()
138 auto make_tuple_cnode = after_fg->NewCNode(make_tuple_inputs); in AddAfterFuncGraph()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/
Dconcat_outputs_for_all_gather.cc92 …std::vector<AnfNodePtr> make_tuple_inputs{NewValueNode(std::make_shared<Primitive>(prim::kPrimMake… in InsertConcatForOutput() local
113 make_tuple_inputs.push_back(concat); in InsertConcatForOutput()
116 auto make_tuple = func_graph->NewCNode(make_tuple_inputs); in InsertConcatForOutput()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/trt_pass/
Dgraph_converter.cc67 std::vector<AnfNodePtr> make_tuple_inputs = {NewValueNode(prim::kPrimMakeTuple)}; in BuildMakeTupleNode() local
92 make_tuple_inputs.push_back(tuple_getitem_cnode); in BuildMakeTupleNode()
97 const CNodePtr &make_tuple_cnode = root->NewCNode(make_tuple_inputs); in BuildMakeTupleNode()
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/
Dauto_monad_eliminate.cc171 auto &make_tuple_inputs = make_tuple->inputs(); in ReplaceLoadUserMakeTuple() local
173 …(void)std::copy_if(make_tuple_inputs.begin(), make_tuple_inputs.end(), std::back_inserter(new_make… in ReplaceLoadUserMakeTuple()
Drecompute.cc391 std::vector<AnfNodePtr> make_tuple_inputs{NewValueNode(prim::kPrimMakeTuple)}; in NewRecomputedNode() local
392 …opy(first_target_inputs.begin(), first_target_inputs.end(), std::back_inserter(make_tuple_inputs)); in NewRecomputedNode()
396 graph->NewCNode(make_tuple_inputs)}; in NewRecomputedNode()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/pass/
Dcommunication_op_fusion.cc296 const auto &make_tuple_inputs = cnode_make_tuple->inputs(); in AdjustAllReduceInputWithLoad() local
297 AnfNodePtrList new_tuple_inputs(make_tuple_inputs.size() - 1); in AdjustAllReduceInputWithLoad()
298 std::copy_if(make_tuple_inputs.cbegin(), make_tuple_inputs.cend(), new_tuple_inputs.begin(), in AdjustAllReduceInputWithLoad()
/third_party/mindspore/mindspore/lite/tools/converter/parser/tf/
Dfunctionalize_while.cc550 std::vector<AnfNodePtr> make_tuple_inputs = tmp_output; in IdentifyBodySubgraphOutput() local
558 make_tuple_inputs.insert(make_tuple_inputs.begin(), make_tuple_prim); in IdentifyBodySubgraphOutput()
559 auto make_tuple_cnode = body_sub_func_graph_->NewCNode(make_tuple_inputs); in IdentifyBodySubgraphOutput()
/third_party/mindspore/tests/ut/cpp/session/
Dkernel_graph_test.cc166 std::vector<AnfNodePtr> make_tuple_inputs = {NewValueNode(prim::kPrimMakeTuple), mul}; in TEST_F() local
167 auto make_tuple = kernel_graph->NewCNode(make_tuple_inputs); in TEST_F()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/
Dascend_helper.cc166 std::vector<AnfNodePtr> make_tuple_inputs = {NewValueNode(prim::kPrimMakeTuple)}; in InsertTransOpForMultipleOutput() local
182 make_tuple_inputs.push_back(trans_op); in InsertTransOpForMultipleOutput()
185 make_tuple_inputs.push_back(tuple_getitem); in InsertTransOpForMultipleOutput()
188 AnfNodePtr make_tuple = func_graph->NewCNode(make_tuple_inputs); in InsertTransOpForMultipleOutput()
/third_party/mindspore/mindspore/lite/tools/converter/parser/caffe/
Dcaffe_model_parser.cc420 std::vector<AnfNodePtr> make_tuple_inputs; in ConvertGraphOutputs() local
425 make_tuple_inputs.emplace_back(make_tuple_prim); in ConvertGraphOutputs()
432 make_tuple_inputs.emplace_back(cnode); in ConvertGraphOutputs()
434 auto make_tuple_cnode = res_graph_->NewCNode(make_tuple_inputs); in ConvertGraphOutputs()
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/pipeline_transformer/
Dpipeline_transformer.cc892 std::vector<AnfNodePtr> make_tuple_inputs; in CutGraph() local
895 if (make_tuple_inputs.empty()) { in CutGraph()
896 make_tuple_inputs = HandleSharedParameter(); in CutGraph()
906 (void)make_tuple_inputs.insert(make_tuple_inputs.end(), send_ops.begin(), send_ops.end()); in CutGraph()
911 auto make_tuple = main_graph_->NewCNode(make_tuple_inputs); in CutGraph()

12