Searched refs:kPrimMulFusion (Results 1 – 11 of 11) sorted by relevance
| /third_party/mindspore/mindspore/lite/tools/optimizer/fusion/ |
| D | tf_gelu_fusion.cc | 76 auto is_mul1 = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimMulFusion>); in DefinePattern()
82 auto is_mul2 = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimMulFusion>); in DefinePattern()
89 auto is_mul3 = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimMulFusion>); in DefinePattern()
92 auto is_mul4 = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimMulFusion>); in DefinePattern()
|
| D | onnx_gelu_fusion.cc | 58 auto is_mul1 = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimMulFusion>); in DefinePattern()
61 auto is_mul2 = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimMulFusion>); in DefinePattern()
|
| D | sigmoid_mul_fusion.cc | 31 auto is_mul = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimMulFusion>); in DefinePattern()
|
| D | norm_fusion.cc | 500 auto is_mul2 = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimMulFusion>); in DefinePattern()
503 auto is_mul1 = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimMulFusion>); in DefinePattern()
506 auto is_mul3 = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimMulFusion>); in DefinePattern()
545 auto is_mul = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimMulFusion>); in DefinePattern()
|
| D | glu_fusion.cc | 75 auto is_mul = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimMulFusion>); in DefinePattern()
|
| D | multi_head_attention_fusion.cc | 56 auto is_mul = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimMulFusion)); in DefineMask()
74 auto is_mul = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimMulFusion)); in DefineMPWithMaskPattern()
|
| D | tflite_rel_pos_multi_head_attention_fusion.cc | 123 auto is_mul = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimMulFusion)); in DefinePatterns()
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/ |
| D | tf_bidirection_gru_cf_fusion.cc | 68 auto mul = VectorRef({GetPrim(prim::kPrimMulFusion), rt, switch3_true}); in DefineGruCellPattern()
76 auto mul2 = VectorRef({GetPrim(prim::kPrimMulFusion), sub, tanh}); in DefineGruCellPattern()
77 auto mul1 = VectorRef({GetPrim(prim::kPrimMulFusion), zt, switch3_true}); in DefineGruCellPattern()
|
| /third_party/mindspore/mindspore/lite/tools/converter/quantizer/ |
| D | quantize_util.cc | 115 … prim::kPrimMulFusion, prim::kPrimReshape, in CanOpFullQuantized()
|
| /third_party/mindspore/mindspore/core/base/ |
| D | core_ops.h | 650 inline const PrimitivePtr kPrimMulFusion = std::make_shared<Primitive>("MulFusion"); variable
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/graph/ |
| D | slice_prepose_pass.cc | 1488 CheckPrimitiveType(preceed_cnode, prim::kPrimMulFusion) || in DoPrepose()
|