Home
last modified time | relevance | path

Searched defs:is_transpose_param (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fusion/
Dflash_attention_fusion_for_custom.cc96 auto is_transpose_param = std::make_shared<CondVar>(IsParamNode); in DefineFlashAttentionPattern2() local
142 auto is_transpose_param = std::make_shared<CondVar>(IsParamNode); in DefineFlashAttentionPattern3() local
Dflash_attention_fusion.cc460 auto is_transpose_param = std::make_shared<Var>(); in DefineFlashAttentionPatternForMsSD21() local
508 auto is_transpose_param = std::make_shared<Var>(); in DefineFlashAttentionPatternForMsSDPseShift() local
708 auto is_transpose_param = std::make_shared<CondVar>(IsParamNode); in DefineFlashAttentionPatternForSDBNSD() local
797 auto is_transpose_param = std::make_shared<CondVar>(IsParamNode); in DefineFlashAttentionPatternForSDBSH() local
965 auto is_transpose_param = std::make_shared<CondVar>(IsParamNode); in DefineFlashAttentionPatternForSDWithoutCast() local