Home
last modified time | relevance | path

Searched defs:make_tuple_node (Results 1 – 25 of 28) sorted by relevance

12

/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fusion/
Dkv_cache_mgr_concat_fusion.cc31 auto make_tuple_node = concat_cnode->input(kInputIndexOne); in GetBatchValidLength() local
Dtf_lstm_cell_fusion.cc179 VectorRef make_tuple_node = VectorRef(outputs); in GetBodyGraphPattern() local
Dtflite_lstm_cell_fusion.cc282 VectorRef make_tuple_node = VectorRef(outputs); in GetBodyGraphPattern() local
Dtf_bidirection_gru_fusion.cc376 VectorRef make_tuple_node = VectorRef(outputs); in GetBodyGraphPattern() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/pass/
Dflatten_concat_fission.cc106 auto make_tuple_node = NewCNode(*concat_nodes, func_graph); in NewMakeTupleNode() local
225 auto make_tuple_node = NewMakeTupleNode(func_graph, &concat_nodes); in Process() local
Dinsert_type_transform_op.cc153 …r CreateRealMakeTupleByMakeTuple(const FuncGraphPtr &func_graph, const CNodePtr &make_tuple_node) { in CreateRealMakeTupleByMakeTuple()
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/graph/
Dinput_and_output_variable_pass.cc106 auto make_tuple_node = make_tuple->cast<CNodePtr>(); in Run() local
Dredundant_op_remove_pass.cc174 auto make_tuple_node = std::make_shared<ops::MakeTuple>(); in ProcessInputHaveDependency() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/optimizer/dynamic_shape/
Dlink_custom_op.cc178 auto make_tuple_node = g->NewCNode(mk_inputs); in AttachDependNodes() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ge/
Dmaketuple_unify_mindir.cc61 auto make_tuple_node = reshape_cnode->input(kIndex2); in Process() local
Dbatchnorm_transform.cc211 auto make_tuple_node = fg->NewCNode({make_tuple_prim, mean_assignsub_node, var_assignsub_node}); in TransformBatchNorm() local
Dlamb_fission.cc284 …auto make_tuple_node = CreateMakeTupleNode(graph, std::vector<AnfNodePtr>{param_node, global_step_… in Process() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/pass/
Doverlap_recompute_allgather_and_flashattention_grad.cc89 auto make_tuple_node = depend_rely_node->cast<CNodePtr>(); in GetDependRealNode() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fusion/
Dshape_reshape_fusion.cc78 auto make_tuple_node = shape->cast<CNodePtr>(); in FindFusionPatternNodes() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/pynative/grad/jit/
Djit_dfunctor.cc154 auto make_tuple_node = output->cast<CNodePtr>(); in GetForwardOutNodeAndBpropGraph() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/graph_util/
Dgraph_splitter.cc120 auto make_tuple_node = func_graph->NewCNode(new_make_tuple_inputs); in CreateMakeTupleNode() local
578 auto make_tuple_node = CreateMakeTupleNode(func_graph, hung_nodes_list); in HandleHungNodes() local
807 auto &make_tuple_node = make_tuple_get_item_nodes.first; in ProcessForSplitOptimizer() local
886 CNodePtr make_tuple_node = CreateNodeWithInterProcessEdgeOnPServer( in CreateNodesForMakeTuple() local
2034 auto make_tuple_node = func_graph_->NewCNode(send_node_tuple_inputs); in AddDependencyBetweenSegments() local
2063 auto make_tuple_node = func_graph_->NewCNode(send_node_tuple_inputs); in AddDependencyBetweenSegments() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fission/
Dlamb_fission.cc218 …auto make_tuple_node = CreateMakeTupleNode(graph, std::vector<AnfNodePtr>{param_node, global_step_… in Process() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/common/
Dcustom_ascend_utils.cc284 auto make_tuple_node = CreateMakeTupleGraphOutput(func_graph, custom_node); in ModifyGraphByCustomNode() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/optimizer/irpass/
Dvmap_eliminate.cc493 auto make_tuple_node = NewValueNode(prim::kPrimMakeTuple); in BindAxis() local
834 const auto make_tuple_node = NewValueNode(prim::kPrimMakeTuple); in BindInAxis() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/core/
Dparallel_op_combine.cc332 auto make_tuple_node = func_graph->NewCNode(mk_inputs); in NewTupleNode() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/cache_embedding/
Dps_embedding_cache_inserter.cc558 auto make_tuple_node = root_graph_->NewCNode(make_tuple_inputs); in ConstructEmbeddingCacheGraph() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/jit/ps/parse/
Dresolve.cc727 auto make_tuple_node = fg->NewCNodeInOrder(inputs); in ResolveSequenceWithAttr() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/jit/ps/static_analysis/
Dauto_monad.cc1775 auto make_tuple_node = func_graph_->NewCNode(tuple_inputs); in HandleNoEliminateNodes() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/pynative/
Dpynative_utils.cc338 auto make_tuple_node = func_graph->NewCNode(inputs); in ConvertValueSequenceToMakeTuple() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/session/
Dkernel_graph_mgr.cc1510 auto make_tuple_node = switch_layer_cnode->input(kSwitchLayerBranchesIndex); in CreateCallSwitchLayerInputs() local

12