Searched defs:reshape3 (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fusion/ |
| D | multi_head_attention_fusion.cc | 263 auto reshape3 = VectorRef({is_reshape3, transpose, var4}); in DefineMPWithMaskPattern() local
308 auto reshape3 = VectorRef({is_reshape3, matmul2, var4}); in DefineMPWithMaskPatternT5() local
378 VectorRef reshape3; in DefineMPWithMaskPatternT5New() local
435 auto reshape3 = VectorRef({is_reshape3, transpose, var4}); in DefineMPWithMaskPatternPA() local
495 auto reshape3 = VectorRef({is_reshape3, transpose, var6}); in DefineMPPatternSwin() local
524 auto reshape3 = VectorRef({is_reshape3, add, var_reshape3}); in DefinePatternPangu() local
|
| D | encoder_layer_fusion.cc | 344 auto reshape3 = VectorRef({is_reshape3, matmul1, var_reshape3}); in DefinePatternMoE() local
559 auto reshape3 = VectorRef({is_reshape3, matmul2, var3}); in DefinePatternEncoderAlpha() local
666 auto reshape3 = VectorRef({is_reshape3, matmul2, var3}); in DefinePatternEncoderLayer() local
|
| D | decoder_layer_fusion.cc | 248 auto reshape3 = VectorRef({is_reshape3, matmul2, var3}); in DefinePatternDecoderLayer() local
|