Searched defs:concat_cnode (Results 1 – 14 of 14) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fusion/ |
| D | kv_cache_mgr_concat_fusion.cc | 30 AnfNodePtr KVCacheMgrConcatFusion::GetBatchValidLength(CNodePtr concat_cnode) { in GetBatchValidLength()
59 auto concat_cnode = kv_cache_cnode->input(kInputIndexThree)->cast<CNodePtr>(); in Run() local
|
| D | add_concat_activation_fusion.cc | 54 auto concat_cnode = concat_node->cast<CNodePtr>(); in Process() local
|
| D | kv_cache_mgr_one_branch_fusion.cc | 125 auto concat_cnode = func_graph->NewCNode(concat_prim_c, {make_tuple_cnode}); in CreateConcatNode() local
|
| D | tensor_dot_fusion.cc | 142 auto concat_cnode = reshape_1_cnode->input(kInputIndexTwo)->cast<CNodePtr>(); in Process() local
|
| D | adjust_matmul_pass.cc | 291 auto concat_cnode = opt::GenConcatNode(func_graph, {shape_gather_node, concat_parm}, in AdjustBMMToMM() local
|
| D | resize_fusion.cc | 221 auto concat_cnode = gather_cnode->input(1)->cast<CNodePtr>(); in DoFuison() local
|
| D | flash_attention_fusion.cc | 179 auto concat_cnode = q_trans_reshape->cast<CNodePtr>()->input(kNumIndex2)->cast<CNodePtr>(); in GetNumHeadForSD() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/pass/ |
| D | split_layernorm_comm_fp.cc | 152 auto concat_cnode = in NewConcatCNodeAndSetAbstract() local
280 const CNodePtr &concat_cnode) { in ExpandSliceRangeToRight()
387 …auto concat_cnode = NewConcatCNodeAndSetAbstract(func_graph, reduce_scatter_a, reduce_scatter_b, k… in SplitIntoInterleaved() local
|
| D | split_matmul_comm_elementwise_fp.cc | 137 auto concat_cnode = func_graph->NewCNode( in SplitIntoInterleaved() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/parallel/ |
| D | operator_info.cc | 164 auto concat_cnode = func_graph_->NewCNode(concat_inputs); in CreateConcateNode() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/optimizer/ |
| D | concat_outputs_for_all_gather.cc | 85 auto concat_cnode = concat->cast<CNodePtr>(); in GenerateKernelBuildInfo() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/graph/ |
| D | padv3_ge_pass.cc | 128 auto concat_cnode = func_graph->NewCNode(concat_prim_c, {make_tuple_cnode}); in CreateConcatNode() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fisson/ |
| D | fisson_util.cc | 373 auto concat_cnode = func_graph->NewCNode(concat_inputs); in CreateOutputsOfConcat() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/ |
| D | step_parallel_utils.cc | 1551 const CNodePtr &concat_cnode, in CanMergeConcatSlice()
1604 … const CNodePtr &concat_cnode, const ShapeVector &concat_output_shape_element, in HandleFuncConcatSlice()
1715 auto concat_cnode = node->cast<CNodePtr>(); in MergeConcatSlice() local
2899 auto concat_cnode = ag_next_nodes.front().first->cast<CNodePtr>(); in InterleavedReplacedConcatNodes() local
|