Home
last modified time | relevance | path

Searched defs:make_tuple_cnode (Results 1 – 25 of 41) sorted by relevance

12

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/enhancer/
Deliminate_maketuple_getitem.cc35 auto make_tuple_cnode = node->cast<CNodePtr>(); in Run() local
Dinsert_depend_for_grad_comm.cc157 auto make_tuple_cnode = cnode_queue_end->input(1)->cast<CNodePtr>(); in IsDxMatMul() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fusion/
Dkv_cache_mgr_concat_fusion.cc33 auto make_tuple_cnode = make_tuple_node->cast<CNodePtr>(); in GetBatchValidLength() local
Dkv_cache_mgr_one_branch_fusion.cc107 auto make_tuple_cnode = in CreateConcatNode() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/optimizer/trt_pass/
Dgraph_converter.cc97 const CNodePtr &make_tuple_cnode = root->NewCNode(make_tuple_inputs); in BuildMakeTupleNode() local
214 const CNodePtr &make_tuple_cnode = BuildMakeTupleNode(root, anf_trt_index_map, trt_node); in ReplaceSubgraphWithTrtNode() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/pass/
Dmerge_cast_opt.cc101 auto make_tuple_cnode = depend_second_input->cast<CNodePtr>(); in MergeCastOpt() local
Dmerge_comm.cc39 auto make_tuple_cnode = make_tuple->cast<CNodePtr>(); in GetMakeTupleValue() local
Dsplit_matmul_comm_elementwise_fp.cc136 …auto make_tuple_cnode = func_graph->NewCNode({NewValueNode(prim::kPrimMakeTuple->Clone()), add_a, … in SplitIntoInterleaved() local
Dsplit_layernorm_comm_fp.cc151 …auto make_tuple_cnode = func_graph->NewCNode({NewValueNode(prim::kPrimMakeTuple->Clone()), input_n… in NewConcatCNodeAndSetAbstract() local
293 auto make_tuple_cnode = concat_cnode->input(kIndex1)->cast<CNodePtr>(); in ExpandSliceRangeToRight() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/optimizer/irpass/
Dpartial_eliminate.h401 auto make_tuple_cnode = switch_layer_cnode->input(kSwitchLayerBranchesIndex)->cast<CNodePtr>(); in operator() local
432 auto make_tuple_cnode = switch_layer->input(kSwitchLayerBranchesIndex)->cast<CNodePtr>(); in BuildNewSwitchLayerNode() local
Dupdatestate_eliminate.cc579 auto make_tuple_cnode = make_tuple->cast<CNodePtr>(); in EliminateUpdateStateBetweenAssignMakeTuple() local
638 auto make_tuple_cnode = make_tuple->cast<CNodePtr>(); in EliminateUpdateStateBetweenMakeTupleAssign() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/adapter/dpico/common/
Dgraph_output_name_keeper.cc49 auto make_tuple_cnode = output_node->cast<api::CNodePtr>(); in SaveOriginalOutputs() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/ops_info/
Dfillv2_info.cc97 auto make_tuple_cnode = dynamic_node->cast<CNodePtr>(); in ReplaceDynamicInput() local
Dtile_info.cc201 auto make_tuple_cnode = multiples_input_node->cast<CNodePtr>(); in UpdateDynamicMultiples() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/graph/
Dadd_tensor_array.cc94 …auto make_tuple_cnode = func_graph->NewCNode({make_tuple_vnode, output_node, tensor_array_write_no… in SetGraphOutput() local
Dpadv3_ge_pass.cc111 auto make_tuple_cnode = func_graph->NewCNode(make_tuple_prim_c, concat_input_vec); in CreateConcatNode() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ge/
Dmaketuple_unify_mindir.cc63 auto make_tuple_cnode = make_tuple_node->cast<CNodePtr>(); in Process() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/parser/third_party/
Dthird_party_model_parser.cc253 auto make_tuple_cnode = func_graph->NewCNode(make_tuple_inputs); in BuildGraphOutputs() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/parser/om/
Dom_model_parser.cc201 auto make_tuple_cnode = func_graph->NewCNode(new_inputs); in CreateMakeTupleGraphOutput() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/import/
Dmindir_control_flow_adjust.cc145 auto make_tuple_cnode = after_fg->NewCNode(make_tuple_inputs); in AddAfterFuncGraph() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/adapter/dpico/parser/caffe/
Dcaffe_model_parser.cc339 auto make_tuple_cnode = res_graph_->NewCNode(make_tuple_inputs); in ConvertGraphOutputs() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/parser/pytorch/
Dpytorch_model_parser.cc239 auto make_tuple_cnode = anf_graph->NewCNode(make_tuple_prim, make_tuple_inputs); in ConvertGraphOutputs() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/c_api/src/
Dgraph.cc162 auto make_tuple_cnode = res_fg->NewCNodeInOrder(out_nodes); in MSFuncGraphSetOutputs() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/parser/tf/
Dfunctionalize_while.cc576 auto make_tuple_cnode = body_sub_func_graph_->NewCNode(make_tuple_inputs); in IdentifyBodySubgraphOutput() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fisson/
Dfisson_util.cc357 auto make_tuple_cnode = func_graph->NewCNode(make_tuple_inputs); in CreateOutputsOfConcat() local

12