Home
last modified time | relevance | path

Searched defs:is_transpose (Results 1 – 9 of 9) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fusion/
Dtranspose_fusion.cc51 auto is_transpose = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimTranspose>); in DefineBNPattern() local
71 auto is_transpose = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimTranspose>); in DefineActivationscalePattern() local
89 auto is_transpose = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimTranspose>); in DefineActivationPattern() local
103 auto is_transpose = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimTranspose>); in DefineBiasAddPattern() local
119 auto is_transpose = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimTranspose>); in DefineScalePattern() local
132 auto is_transpose = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimTranspose>); in DefineSoftmaxPattern() local
Dflash_attention_fusion_for_custom.cc98 auto is_transpose = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimTranspose>); in DefineFlashAttentionPattern2() local
144 auto is_transpose = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimTranspose>); in DefineFlashAttentionPattern3() local
Dreshape_transpose_fusion.cc42 auto is_transpose = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimTranspose)); in DefineReshapeTransposePattern() local
52 auto is_transpose = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimTranspose)); in DefineTransposeReshapePattern() local
Dmulti_head_attention_fusion.cc254 …auto is_transpose = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimTranspose), "is_t… in DefineMPWithMaskPattern() local
380 …auto is_transpose = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimTranspose), "tran… in DefineMPWithMaskPatternT5New() local
426 auto is_transpose = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimTranspose)); in DefineMPWithMaskPatternPA() local
486 auto is_transpose = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimTranspose)); in DefineMPPatternSwin() local
537 …auto is_transpose = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimTranspose), "is_t… in DefinePatternPangu() local
Dconv_pad_fusion.cc203 auto is_transpose = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimTranspose>); in DefinePadTransposeConvPattern() local
Dtflite_rel_pos_multi_head_attention_fusion.cc516 auto is_transpose = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimTranspose)); in DefineProcessInputPattern() local
528 auto is_transpose = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimTranspose)); in DefineProcessOutputPattern() local
Dresize_fusion.cc102 auto is_transpose = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimTranspose>); in DefinePattern() local
Dtf_bidirection_gru_fusion.cc189 auto is_transpose = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimTranspose)); in DefineFowardPattern() local
211 auto is_transpose = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimTranspose)); in DefinebackwardPattern() local
Dflash_attention_fusion.cc462 auto is_transpose = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimTranspose>); in DefineFlashAttentionPatternForMsSD21() local
510 auto is_transpose = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimTranspose>); in DefineFlashAttentionPatternForMsSDPseShift() local
710 auto is_transpose = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimTranspose>); in DefineFlashAttentionPatternForSDBNSD() local
799 auto is_transpose = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimTranspose>); in DefineFlashAttentionPatternForSDBSH() local
967 auto is_transpose = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimTranspose>); in DefineFlashAttentionPatternForSDWithoutCast() local