Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fusion/
Dflash_attention_fusion.cc856 auto is_transpose_o = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimTranspose>); in DefineFlashAttentionPatternForSDBSH() local
1018 auto is_transpose_o = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimTranspose>); in DefineFlashAttentionPatternForSDWithoutCast() local
1288 auto is_transpose_o = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimTranspose>); in DefineFlashAttentionPatternForSDEinsum() local