Home
last modified time | relevance | path

Searched defs:pattern_name (Results 1 – 25 of 29) sorted by relevance

12

/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fusion/
Dsigmoid_mul_fusion.cc60 AnfNodePtr SigmoidMulFusion::Process(const std::string &pattern_name, const mindspore::FuncGraphPtr… in Process()
85 bool SigmoidMulFusion::CheckPattern(const std::string &pattern_name, const FuncGraphPtr &func_graph, in CheckPattern()
Dgelu_fusion.cc72 AnfNodePtr GeLUFusion::Process(const std::string &pattern_name, const mindspore::FuncGraphPtr &func… in Process()
Dconv_pad_fusion.cc51 const std::string &pattern_name) { in ReplaceParamsAndNodes()
226 AnfNodePtr ConvPadFusion::Process(const std::string &pattern_name, const FuncGraphPtr &func_graph, in Process()
Dgroupnormsilu_fusion.cc318 CNodePtr GroupNormSiluFusion::CreateGroupNormSiluNodeForSD15(const std::string &pattern_name, in CreateGroupNormSiluNodeForSD15()
365 CNodePtr GroupNormSiluFusion::CreateGroupNormSiluNodeForSDWithCast(const std::string &pattern_name, in CreateGroupNormSiluNodeForSDWithCast()
418 …dePtr GroupNormSiluFusion::CreateGroupNormSiluNodeForSDWithoutSilu(const std::string &pattern_name, in CreateGroupNormSiluNodeForSDWithoutSilu()
Donnx_gelu_fusion.cc102 bool OnnxGeLUFusion::CheckPattern(const std::string &pattern_name, const EquivPtr &equiv) const { in CheckPattern() argument
Dtf_gelu_fusion.cc106 bool TfGeLUFusion::CheckPattern(const std::string &pattern_name, const EquivPtr &equiv) const { in CheckPattern() argument
Dffn_fusion.cc233 bool FFNFusion::CheckPattern(const std::string &pattern_name, const EquivPtr &equiv) const { in CheckPattern() argument
275 AnfNodePtr FFNFusion::Process(const std::string &pattern_name, const mindspore::FuncGraphPtr &func_… in Process()
Dfullconnected_add_fusion.cc142 AnfNodePtr FullconnectedAddFusion::Process(const std::string &pattern_name, const FuncGraphPtr &fun… in Process()
Dflash_attention_fusion.cc1751const std::string &pattern_name, const FuncGraphPtr &func_graph, const AnfNodePtr &node, const Equ… in CreateFlashAttentionNodeForMsSDXL()
1804const std::string &pattern_name, const FuncGraphPtr &func_graph, const AnfNodePtr &node, const Equ… in CreateFlashAttentionNodeForMsSDPseShift()
1926const std::string &pattern_name, const FuncGraphPtr &func_graph, const AnfNodePtr &node, const Equ… in CreateFlashAttentionNodeForMsSD21()
1988const std::string &pattern_name, const FuncGraphPtr &func_graph, const AnfNodePtr &node, const Equ… in CreateFlashAttentionNodeForVideoComposer()
2079 CNodePtr FlashAttentionFusion::CreateFlashAttentionNodeForSD(const std::string &pattern_name, in CreateFlashAttentionNodeForSD()
2170const std::string &pattern_name, const FuncGraphPtr &func_graph, const AnfNodePtr &node, const Equ… in CreateFlashAttentionNodeForSDPreMul()
2263const std::string &pattern_name, const FuncGraphPtr &func_graph, const AnfNodePtr &node, const Equ… in CreateFlashAttentionNodeForSDWithoutCast()
2370const std::string &pattern_name, const FuncGraphPtr &func_graph, const AnfNodePtr &node, const Equ… in CreateFlashAttentionNodeForPanGu()
2440const std::string &pattern_name, const FuncGraphPtr &func_graph, const AnfNodePtr &node, const Equ… in CreateFlashAttentionNodeForLLAMAPatternV1()
2475const std::string &pattern_name, const FuncGraphPtr &func_graph, const AnfNodePtr &node, const Equ… in CreateFlashAttentionNodeForLLAMAPatternV2()
[all …]
Dflash_attention_fusion_for_custom.cc216 …dePtr FlashAttentionFusionForCustom::CreateFlashAttentionNodePart1(const std::string &pattern_name, in CreateFlashAttentionNodePart1()
295 …dePtr FlashAttentionFusionForCustom::CreateFlashAttentionNodePart2(const std::string &pattern_name, in CreateFlashAttentionNodePart2()
Dmul_add_fusion.cc225 AnfNodePtr MulAddFusion::Process(const std::string &pattern_name, const mindspore::FuncGraphPtr &fu… in Process()
Dreshape_reshape_fusion.cc168 AnfNodePtr ReshapeReshapeFusion::Process(const std::string &pattern_name, const FuncGraphPtr &func_… in Process()
Dcast_fusion.cc333 AnfNodePtr CastFusionPass::Process(const std::string &pattern_name, const mindspore::FuncGraphPtr &… in Process()
Dtranspose_fusion.cc322 AnfNodePtr TransposeFusion::Process(const std::string &pattern_name, const mindspore::FuncGraphPtr … in Process()
Dmatmul_allreduce_fusion.cc359 AnfNodePtr MatMulAllReduceFusion::Process(const std::string &pattern_name, const mindspore::FuncGra… in Process()
Dadd_layernorm_fusion.cc265 AnfNodePtr LayerNormV3Fusion::Process(const std::string &pattern_name, const mindspore::FuncGraphPt… in Process()
Dreshape_transpose_fusion.cc434 AnfNodePtr ReshapeTransposeFusion::Process(const std::string &pattern_name, const mindspore::FuncGr… in Process()
Ddecoder_layer_fusion.cc275 AnfNodePtr DecoderLayerFusion::Process(const std::string &pattern_name, const mindspore::FuncGraphP… in Process()
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/train/optimizer/fusion/
Dmatmul_activation_fusion_pass.cc51 MetaGraphT *graph, const std::string &pattern_name, in DoFusion()
Dreshape_gather_reshape_fusion_pass.cc74 MetaGraphT *graph, const std::string &pattern_name, in DoFusion()
Dmatmul_matmul_add_fusion_pass.cc89 STATUS MatMulMatMulAddFusionPass::DoFusion(MetaGraphT *graph, const std::string &pattern_name, in DoFusion()
Dmatmul_add_fusion_pass.cc76 STATUS MatMulAddFusionPass::DoFusion(MetaGraphT *graph, const std::string &pattern_name, in DoFusion()
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/legacy_optimizer/fusion/
Dmatmul_biasadd_fusion_pass.cc53 STATUS MatMulBiasAddFusionPass::DoFusion(MetaGraphT *graph, const std::string &pattern_name, in DoFusion()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fusion/
Dinference_matmul_split_fusion.cc52 std::string pattern_name = ""; in Run() local
78 std::string pattern_name = ""; in GetSplitFusionPatternName() local
112 std::string pattern_name = ""; in GetFusionPatternName() local
523 AnfNodePtr InferenceMatmulSplitFusion::Process(const std::string &pattern_name, const FuncGraphPtr … in Process()
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/quantizer/
Dcle_pattern.cc116 AnfNodePtr CLEPattern::Process(const string &pattern_name, const FuncGraphPtr &func_graph, const An… in Process()

12