Home
last modified time | relevance | path

Searched defs:make_tuple_inputs (Results 1 – 25 of 59) sorted by relevance

123

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fission/
Dscale_grad_fission.cc89 std::vector<AnfNodePtr> make_tuple_inputs = {NewValueNode(prim::kPrimMakeTuple)}; in Process() local
117 std::vector<AnfNodePtr> make_tuple_inputs = {NewValueNode(prim::kPrimMakeTuple)}; in Process() local
Dsplit_fission.cc130 std::vector<AnfNodePtr> make_tuple_inputs = {NewValueNode(prim::kPrimMakeTuple)}; in DoFission() local
Dbn_grad_split.cc134 …std::vector<AnfNodePtr> make_tuple_inputs = {NewValueNode(prim::kPrimMakeTuple), bn_reduce_grad_ou… in BNGradSplitForTBE() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/optimizer/
Dinsert_cast_cpu.cc58 std::vector<AnfNodePtr> *make_tuple_inputs) { in SyncWeightNodeWithCast()
77 …std::vector<AnfNodePtr> make_tuple_inputs{NewValueNode(std::make_shared<Primitive>(prim::kPrimMake… in InsertCast() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/enhancer/
Dinsert_tensor_move_for_getnext.cc39 std::vector<AnfNodePtr> make_tuple_inputs; in InsertTensorMoveForGetNextOutputs() local
Dinsert_depend_for_all_gather.cc61 std::vector<AnfNodePtr> make_tuple_inputs{NewValueNode(prim::kPrimMakeTuple)}; in Run() local
Dinsert_depend_for_grad_comm.cc331 std::vector<AnfNodePtr> make_tuple_inputs{NewValueNode(prim::kPrimMakeTuple)}; in Run() local
368 std::vector<AnfNodePtr> make_tuple_inputs{NewValueNode(prim::kPrimMakeTuple)}; in Run() local
Dconcat_outputs_for_all_gather.cc167 …std::vector<AnfNodePtr> make_tuple_inputs{NewValueNode(std::make_shared<Primitive>(prim::kPrimMake… in InsertConcatForOutput() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/format_type/
Dinsert_transpose_for_basiclstm_op.cc75 std::vector<AnfNodePtr> make_tuple_inputs = {NewValueNode(prim::kPrimMakeTuple)}; in Insert() local
Dinsert_cast.cc41 std::vector<AnfNodePtr> make_tuple_inputs; in InsertCastForMultipleOutput() local
Ddeal_ref_output.cc192 std::vector<AnfNodePtr> make_tuple_inputs; in DealRefForMultipleOutput() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/pass/
Dmerge_cast_opt.cc91 std::vector<AnfNodePtr> make_tuple_inputs{NewValueNode(prim::kPrimMakeTuple)}; in MergeCastOpt() local
Doverlap_opt_shard_in_pipeline.cc134 std::vector<AnfNodePtr> make_tuple_inputs{NewValueNode(prim::kPrimMakeTuple)}; in OverlapOptShardInPipeline() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/adapter/dpico/common/
Dgraph_output_name_keeper.cc51 auto make_tuple_inputs = make_tuple_cnode->inputs(); in SaveOriginalOutputs() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/parser/pytorch/
Dpytorch_model_parser.cc218 std::vector<AnfNodePtr> make_tuple_inputs; in ConvertGraphOutputs() local
339 std::vector<AnfNodePtr> make_tuple_inputs; in ConvertConstNode() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fusion/
Dbatchnorm_to_bninfer.cc163 std::vector<AnfNodePtr> make_tuple_inputs = { in Process() local
/third_party/mindspore/mindspore-src/source/tests/ut/cpp/session/
Dkernel_graph_test.cc169 std::vector<AnfNodePtr> make_tuple_inputs = {NewValueNode(prim::kPrimMakeTuple), mul}; in TEST_F() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/mindir/
Ddropout_unify_mindir.cc367 std::vector<AnfNodePtr> make_tuple_inputs; in UpdateReturnNode() local
609 …std::vector<AnfNodePtr> make_tuple_inputs{NewValueNode(prim::kPrimMakeTuple), dropout_do_mask, dro… in Process() local
672 …std::vector<AnfNodePtr> make_tuple_inputs{NewValueNode(prim::kPrimMakeTuple), dropout_do_mask, dro… in Process() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/parser/third_party/
Dthird_party_model_parser.cc251 std::vector<AnfNodePtr> make_tuple_inputs = output_nodes; in BuildGraphOutputs() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/optimizer/
Dconcat_outputs_for_all_gather.cc106 …std::vector<AnfNodePtr> make_tuple_inputs{NewValueNode(std::make_shared<Primitive>(prim::kPrimMake… in InsertConcatForOutput() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/optimizer/irpass/
Dupdatestate_eliminate.cc352 const AnfNodePtrList &make_tuple_inputs) { in MakeTupleForSameNodes()
447 AnfNodePtrList make_tuple_inputs; in EliminateUpdateStateForLoads() local
515 AnfNodePtrList make_tuple_inputs{NewValueNode(prim::kPrimMakeTuple), a1, assign}; in EliminateUpdateStateBetweenAssigns() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/optimizer/trt_pass/
Dgraph_converter.cc67 std::vector<AnfNodePtr> make_tuple_inputs = {NewValueNode(prim::kPrimMakeTuple)}; in BuildMakeTupleNode() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ge/
Ddropout_for_ge.cc174 …std::vector<AnfNodePtr> make_tuple_inputs = {NewValueNode(prim::kPrimMakeTuple), dropout_do_mask_n… in Process() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/import/
Dmindir_control_flow_adjust.cc134 std::vector<AnfNodePtr> make_tuple_inputs = after_fg->get_inputs(); in AddAfterFuncGraph() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/pipeline_transformer/
Dgpipe_interleave_scheduler.cc306 std::vector<AnfNodePtr> make_tuple_inputs = {NewValueNode(prim::kPrimMakeTuple)}; in OptimizerShardCommReorder() local

123