Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fusion/
Dflash_attention_fusion.cc762 …auto input_q_reshape = VectorRef({is_input_q_reshape, input_q_reshape_param_1, input_q_reshape_par… in DefineFlashAttentionPatternForSDBSH() local
930 …auto input_q_reshape = VectorRef({is_input_q_reshape, input_q_reshape_param_1, input_q_reshape_par… in DefineFlashAttentionPatternForSDWithoutCast() local