Searched defs:is_reshape3 (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fusion/ |
| D | multi_head_attention_fusion.cc | 259 …auto is_reshape3 = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimReshape), "is_resh… in DefineMPWithMaskPattern() local
304 …auto is_reshape3 = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimReshape), "reshape… in DefineMPWithMaskPatternT5() local
374 …auto is_reshape3 = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimReshape), "reshape… in DefineMPWithMaskPatternT5New() local
431 …auto is_reshape3 = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimReshape), "is_resh… in DefineMPWithMaskPatternPA() local
491 …auto is_reshape3 = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimReshape), "is_resh… in DefineMPPatternSwin() local
520 …auto is_reshape3 = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimReshape), "is_resh… in DefinePatternPangu() local
|
| D | encoder_layer_fusion.cc | 340 …auto is_reshape3 = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimReshape), "is_resh… in DefinePatternMoE() local
555 …auto is_reshape3 = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimReshape), "reshape… in DefinePatternEncoderAlpha() local
662 …auto is_reshape3 = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimReshape), "reshape… in DefinePatternEncoderLayer() local
|
| D | decoder_layer_fusion.cc | 244 …auto is_reshape3 = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimReshape), "reshape… in DefinePatternDecoderLayer() local
|
| D | tflite_rel_pos_multi_head_attention_fusion.cc | 512 auto is_reshape3 = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimReshape)); in DefineProcessInputPattern() local
|