Searched defs:split_cnode (Results 1 – 7 of 7) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fusion/ |
| D | inference_matmul_split_fusion.cc | 174 PrimitivePtr InferenceMatmulSplitFusion::CreateMatmulSplitPrim(const CNodePtr &split_cnode, size_t … in CreateMatmulSplitPrim()
197 auto split_cnode = node->cast<CNodePtr>(); in CreateMatmulSplitNode() local
238 auto split_cnode = node->cast<CNodePtr>(); in CreateMatmulBiasAddSplitNode() local
283 auto split_cnode = node->cast<CNodePtr>(); in CreateQuantbatchmatmulSplitNode() local
322 …ceMatmulSplitFusion::CreateGetItemNode(const FuncGraphPtr &func_graph, const CNodePtr &split_cnode, in CreateGetItemNode()
369 auto split_cnode = item_cnode->input(kIndex1)->cast<CNodePtr>(); in CreateMatmulSplitSiluNode() local
420 auto split_cnode = get_item_cnode->input(kIndex1)->cast<CNodePtr>(); in CreateMatmulBiasAddSplitSiluNode() local
477 auto split_cnode = item_cnode->input(kIndex1)->cast<CNodePtr>(); in CreateQuantbatchmatmulSplitSiluNode() local
530 auto split_cnode = node->cast<CNodePtr>(); in Process() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/pass/ |
| D | split_layernorm_comm_fp.cc | 134 auto split_cnode = func_graph->NewCNode({NewValueNode(prim::kPrimSplit->Clone()), input_node, in NewSplitCNodeAndSetAbstract() local
237 const CNodePtr &split_cnode) { in ExpandSliceRangeToLeft()
328 …auto split_cnode = NewSplitCNodeAndSetAbstract(func_graph, layernorm_cnode->input(kIndex1), kLongZ… in SplitIntoInterleaved() local
|
| D | split_matmul_comm_elementwise_fp.cc | 112 auto split_cnode = func_graph->NewCNode({NewValueNode(prim::kPrimSplit->Clone()), matmul_input1, in SplitIntoInterleaved() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fisson/ |
| D | eliminate_concat_split.cc | 168 auto split_cnode = node->cast<CNodePtr>(); in Process() local
|
| D | fisson_util.cc | 411 auto split_cnode = func_graph->NewCNode(split_inputs); in CreateOutputsOfSplitWithOverlap() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/parallel/ |
| D | conv2d_info.cc | 196 auto split_cnode = func_graph_->NewCNode(split_inputs); in CreateOutputsOfSplit() local
|
| D | depthwise_conv2d_info.cc | 394 auto split_cnode = func_graph_->NewCNode(split_prim_c, split_inputs); in CreateOutputsOfSplit() local
|