Home
last modified time | relevance | path

Searched refs:kPrimSubFusion (Results 1 – 6 of 6) sorted by relevance

/third_party/mindspore/mindspore/lite/tools/optimizer/fusion/
Dnorm_fusion.cc509 auto is_sub = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimSubFusion>); in DefinePattern()
524 auto is_sub1 = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimSubFusion>); in DefinePattern()
527 auto is_sub2 = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimSubFusion>); in DefinePattern()
Dmulti_head_attention_fusion.cc51 auto is_sub = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimSubFusion)); in DefineMask()
/third_party/mindspore/mindspore/lite/tools/converter/parser/
Dtf_bidirection_gru_cf_fusion.cc75 …auto sub = VectorRef({GetPrim(prim::kPrimSubFusion), std::make_shared<CondVar>(IsParameterNode), z… in DefineGruCellPattern()
/third_party/mindspore/mindspore/lite/tools/converter/quantizer/
Dquantize_util.cc119 … prim::kPrimSubFusion, prim::kPrimUnsqueeze, in CanOpFullQuantized()
/third_party/mindspore/mindspore/core/base/
Dcore_ops.h649 inline const PrimitivePtr kPrimSubFusion = std::make_shared<Primitive>("SubFusion"); variable
/third_party/mindspore/mindspore/lite/tools/optimizer/graph/
Dslice_prepose_pass.cc1487 } else if (CheckPrimitiveType(preceed_cnode, prim::kPrimSubFusion) || in DoPrepose()