Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fusion/
Dflash_attention_fusion.cc789 …auto input_k_transpose = VectorRef({is_input_k_transpose, input_k_reshape, is_input_k_transpose_pa… in DefineFlashAttentionPatternForSDBSH() local
957 …auto input_k_transpose = VectorRef({is_input_k_transpose, input_k_reshape, is_input_k_transpose_pa… in DefineFlashAttentionPatternForSDWithoutCast() local