Searched defs:input_q_transpose (Results 1 – 1 of 1) sorted by relevance
768 …auto input_q_transpose = VectorRef({is_input_q_transpose, input_q_reshape, is_input_q_transpose_pa… in DefineFlashAttentionPatternForSDBSH() local936 …auto input_q_transpose = VectorRef({is_input_q_transpose, input_q_reshape, is_input_q_transpose_pa… in DefineFlashAttentionPatternForSDWithoutCast() local