Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fusion/
Dflash_attention_fusion.cc864 auto reshape_o2 = VectorRef({is_reshape_o2, transpose_o, reshape_o2_2}); in DefineFlashAttentionPatternForSDBSH() local
1026 auto reshape_o2 = VectorRef({is_reshape_o2, transpose_o, reshape_o2_2}); in DefineFlashAttentionPatternForSDWithoutCast() local
1296 auto reshape_o2 = VectorRef({is_reshape_o2, transpose_o, reshape_o2_2}); in DefineFlashAttentionPatternForSDEinsum() local
2084 auto reshape_o2 = cnode; in CreateFlashAttentionNodeForSD() local
2268 auto reshape_o2 = cnode; in CreateFlashAttentionNodeForSDWithoutCast() local
2539 auto reshape_o2 = cnode; in CreateFlashAttentionNodeForSDEinsum() local