Home
last modified time | relevance | path

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

12

/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
133 …std::vector<AnfNodePtr> make_tuple_inputs = {NewValueNode(prim::kPrimMakeTuple), bn_reduce_grad_ou… in SyncBNGradSplitForTBE() local
Dbatch_norm_grad_split.cc125 …std::vector<AnfNodePtr> make_tuple_inputs = {NewValueNode(prim::kPrimMakeTuple), bn_reduce_grad_ou… in Process() local
Dlayer_norm_grad_split.cc127 …std::vector<AnfNodePtr> make_tuple_inputs = {NewValueNode(prim::kPrimMakeTuple), layer_norm_x_back… in Process() local
Dbatch_norm_grad_infer_fission.cc177 std::vector<AnfNodePtr> make_tuple_inputs = {NewValueNode(prim::kPrimMakeTuple), bn_infer_grad, in Process() local
Dsplit_fission.cc138 std::vector<AnfNodePtr> make_tuple_inputs = {NewValueNode(prim::kPrimMakeTuple)}; in DoFission() local
Ddynamic_gru_v2_grad_fission.cc524 std::vector<AnfNodePtr> make_tuple_inputs = {NewValueNode(prim::kPrimMakeTuple), in Process() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/enhancer/
Dinsert_tensor_move_for_getnext.cc37 std::vector<AnfNodePtr> make_tuple_inputs; in InsertTensorMoveForGetNextOutputs() local
Dconcat_outputs_for_all_gather.cc94 …std::vector<AnfNodePtr> make_tuple_inputs{NewValueNode(std::make_shared<Primitive>(prim::kPrimMake… in InsertConcatForOutput() local
/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
213 …std::vector<AnfNodePtr> make_tuple_inputs = {NewValueNode(prim::kPrimMakeTuple), lsq_perchannel_gr… in Process() local
Ddropout_unify_mindir.cc315 …std::vector<AnfNodePtr> make_tuple_inputs{NewValueNode(prim::kPrimMakeTuple), dropout_do_mask, dro… in Process() local
355 …std::vector<AnfNodePtr> make_tuple_inputs{NewValueNode(prim::kPrimMakeTuple), dropout_do_mask, dro… in Process() local
/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
Dinsert_cast.cc43 std::vector<AnfNodePtr> make_tuple_inputs; in InsertCastForMultipleOutput() local
Ddeal_ref_and_split_unsupported_transdata.cc184 std::vector<AnfNodePtr> make_tuple_inputs; in DealRefForMultipleOutput() local
/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
/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
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/
Dupdatestate_eliminate.cc254 const AnfNodePtrList &make_tuple_inputs) { in MakeTupleForSameNodes()
349 AnfNodePtrList make_tuple_inputs; in EliminateUpdateStateForLoads() local
417 AnfNodePtrList make_tuple_inputs{NewValueNode(prim::kPrimMakeTuple), a1, assign}; in EliminateUpdateStateBetweenAssigns() local
/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
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/trt_pass/
Dgraph_converter.cc67 std::vector<AnfNodePtr> make_tuple_inputs = {NewValueNode(prim::kPrimMakeTuple)}; in BuildMakeTupleNode() local
/third_party/mindspore/mindspore/ccsrc/frontend/optimizer/
Dauto_monad_eliminate.cc171 auto &make_tuple_inputs = make_tuple->inputs(); in ReplaceLoadUserMakeTuple() local
Drecompute.cc391 std::vector<AnfNodePtr> make_tuple_inputs{NewValueNode(prim::kPrimMakeTuple)}; in NewRecomputedNode() local
/third_party/mindspore/mindspore/lite/tools/converter/parser/tf/
Dfunctionalize_while.cc550 std::vector<AnfNodePtr> make_tuple_inputs = tmp_output; in IdentifyBodySubgraphOutput() local
/third_party/mindspore/mindspore/lite/tools/converter/parser/caffe/
Dcaffe_model_parser.cc420 std::vector<AnfNodePtr> make_tuple_inputs; in ConvertGraphOutputs() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/pass/
Dcommunication_op_fusion.cc296 const auto &make_tuple_inputs = cnode_make_tuple->inputs(); in AdjustAllReduceInputWithLoad() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/
Dascend_helper.cc166 std::vector<AnfNodePtr> make_tuple_inputs = {NewValueNode(prim::kPrimMakeTuple)}; in InsertTransOpForMultipleOutput() local
/third_party/mindspore/mindspore/ccsrc/backend/session/
Dkernel_graph.cc637 std::vector<AnfNodePtr> make_tuple_inputs = { in TransValueNodeTuple() local
655 std::vector<AnfNodePtr> make_tuple_inputs = { in TransParameterTuple() local

12