Searched refs:kPrimAddFusion (Results 1 – 14 of 14) sorted by relevance
| /third_party/mindspore/mindspore/lite/test/ut/tools/converter/registry/ |
| D | model_parser_registry_test.cc | 52 bool is_add = opt::CheckPrimitiveType(*iter, prim::kPrimAddFusion); in TEST_F()
55 is_add = opt::CheckPrimitiveType(*iter, prim::kPrimAddFusion); in TEST_F()
|
| D | pass_registry_test.cc | 59 if (!opt::CheckPrimitiveType(cnode, prim::kPrimAddFusion)) { in CanFusion()
75 if (!opt::CheckPrimitiveType(input, prim::kPrimAddFusion)) { in CanFusion()
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/fusion/ |
| D | tf_gelu_fusion.cc | 79 auto is_add1 = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimAddFusion>); in DefinePattern()
86 auto is_add2 = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimAddFusion>); in DefinePattern()
|
| D | tflite_rel_pos_multi_head_attention_fusion.cc | 83 auto is_add1 = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimAddFusion)); in DefinePatterns()
98 auto is_add2 = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimAddFusion)); in DefinePatterns()
120 auto is_add3 = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimAddFusion)); in DefinePatterns()
128 auto is_add4 = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimAddFusion)); in DefinePatterns()
396 auto is_add = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimAddFusion)); in DefineProcessInputPattern()
439 auto is_add = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimAddFusion)); in DefineProcessOutputPattern()
|
| D | matmul_add_fusion.cc | 58 …if (!CheckPrimitiveType(node, prim::kPrimAddFusion) && !CheckPrimitiveType(node, prim::kPrimBiasAd… in Run()
|
| D | onnx_gelu_fusion.cc | 55 auto is_add = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimAddFusion>); in DefinePattern()
|
| D | norm_fusion.cc | 494 auto is_add1 = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimAddFusion>); in DefinePattern()
512 auto is_add2 = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimAddFusion>); in DefinePattern()
536 auto is_add1 = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimAddFusion>); in DefinePattern()
548 auto is_add2 = std::make_shared<CondVar>(IsSpecifiedNode<&prim::kPrimAddFusion>); in DefinePattern()
|
| D | conv_biasadd_fusion.cc | 39 …return CheckPrimitiveType(anf_node, prim::kPrimAddFusion) || CheckPrimitiveType(anf_node, prim::kP… in IsAddNode()
|
| D | multi_head_attention_fusion.cc | 81 auto is_add = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimAddFusion)); in DefineMPWithMaskPattern()
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/ |
| D | tf_bidirection_gru_cf_fusion.cc | 78 auto add = VectorRef({GetPrim(prim::kPrimAddFusion), mul1, mul2}); in DefineGruCellPattern()
98 …auto fw_add = VectorRef({GetPrim(prim::kPrimAddFusion), fw_switch_true, std::make_shared<CondVar>(… in DefineBidirectionRnnPattern()
116 …VectorRef({GetPrim(prim::kPrimAddFusion), std::make_shared<SeqVar>(), std::make_shared<CondVar>(Is… in DefineBidirectionRnnPattern()
|
| /third_party/mindspore/mindspore/lite/examples/converter_extend/src/ |
| D | pass_registry_tutorial.cc | 84 if (!CheckPrimitiveTypeTutorial(node, prim::kPrimAddFusion)) { in Execute()
|
| /third_party/mindspore/mindspore/lite/tools/converter/quantizer/ |
| D | quantize_util.cc | 109 …static const std::set<PrimitivePtr> support_int8_ops = {prim::kPrimAddFusion, prim::kPrimActi… in CanOpFullQuantized()
|
| /third_party/mindspore/mindspore/core/base/ |
| D | core_ops.h | 647 inline const PrimitivePtr kPrimAddFusion = std::make_shared<Primitive>("AddFusion"); variable
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/graph/ |
| D | slice_prepose_pass.cc | 1489 CheckPrimitiveType(preceed_cnode, prim::kPrimAddFusion)) { in DoPrepose()
|