From 23b9883716c034e4829de521b8949c8e89e1e0c8 Mon Sep 17 00:00:00 2001 From: qnqinan Date: Tue, 14 Aug 2018 17:54:04 +0800 Subject: [PATCH] add fusion conv bn op type --- src/common/types.cpp | 4 +++- src/common/types.h | 1 + 2 files changed, 4 insertions(+), 1 deletion(-) diff --git a/src/common/types.cpp b/src/common/types.cpp index b638750385..41bbfa5256 100644 --- a/src/common/types.cpp +++ b/src/common/types.cpp @@ -50,6 +50,7 @@ const char *G_OP_TYPE_FUSION_ELEMENTWISE_ADD_RELU = "fusion_elementwise_add_relu"; const char *G_OP_TYPE_FUSION_FC_RELU = "fusion_fc_relu"; const char *G_OP_TYPE_REGION = "region"; +const char *G_OP_TYPE_FUSION_CONV_BN = "fusion_conv_bn"; std::unordered_map< std::string, std::pair, std::vector>> @@ -85,6 +86,7 @@ std::unordered_map< {G_OP_TYPE_FUSION_POOL_BN, {{"X"}, {"Y"}}}, {G_OP_TYPE_FUSION_ELEMENTWISE_ADD_RELU, {{"X", "Y"}, {"Out"}}}, {G_OP_TYPE_FUSION_FC_RELU, {{"X", "Y", "Z"}, {"Out"}}}, - {G_OP_TYPE_REGION, {{"X"}, {"Out"}}}}; + {G_OP_TYPE_REGION, {{"X"}, {"Out"}}}, + {G_OP_TYPE_FUSION_CONV_BN, {{"Input"}, {"Y"}}}}; } // namespace paddle_mobile diff --git a/src/common/types.h b/src/common/types.h index 6066879305..78c96f327a 100644 --- a/src/common/types.h +++ b/src/common/types.h @@ -113,6 +113,7 @@ extern const char *G_OP_TYPE_FUSION_POOL_BN; extern const char *G_OP_TYPE_FUSION_ELEMENTWISE_ADD_RELU; extern const char *G_OP_TYPE_FUSION_FC_RELU; extern const char *G_OP_TYPE_REGION; +extern const char *G_OP_TYPE_FUSION_CONV_BN; extern std::unordered_map< std::string, std::pair, std::vector>> -- GitLab