diff --git a/src/common/types.cpp b/src/common/types.cpp index 7b8b5bb559a29bd28c9789f3b52fa1b65cc14dc5..e64d9c91a6faac2b7f2eaccac35b7592ab445efc 100644 --- a/src/common/types.cpp +++ b/src/common/types.cpp @@ -44,6 +44,14 @@ const std::string G_OP_TYPE_FETCH = "fetch"; const std::string G_OP_TYPE_DEPTHWISE_CONV = "depthwise_conv2d"; const std::string G_OP_TYPE_IM2SEQUENCE = "im2sequence"; const std::string G_OP_TYPE_DROPOUT = "dropout"; +const std::string G_OP_TYPE_FUSION_CONV_RELU = "fusion_conv_relu"; +const std::string G_OP_TYPE_FUSION_CONV_BN_SCALE = "fusion_conv_bn_scale"; +const std::string G_OP_TYPE_FUSION_CONV_BN_SCALE_RELU = + "fusion_conv_bn_scale_relu"; +const std::string G_OP_TYPE_FUSION_POOL_BN = "fusion_pool_bn"; +const std::string G_OP_TYPE_FUSION_ELEMENTWISE_ADD_RELU = + "fusion_elementwise_add_relu"; +const std::string G_OP_TYPE_REGION = "region"; std::unordered_map< std::string, std::pair, std::vector>> @@ -74,6 +82,12 @@ std::unordered_map< {G_OP_TYPE_DEPTHWISE_CONV, {{"Input"}, {"Output"}}}, {G_OP_TYPE_FUSION_CONV_ADD_RELU, {{"Input"}, {"Out"}}}, {G_OP_TYPE_IM2SEQUENCE, {{"X"}, {"Out"}}}, - {G_OP_TYPE_DROPOUT, {{"X"}, {"Out"}}}}; + {G_OP_TYPE_DROPOUT, {{"X"}, {"Out"}}}, + {G_OP_TYPE_FUSION_CONV_RELU, {{"Input"}, {"Out"}}}, + {G_OP_TYPE_FUSION_CONV_BN_SCALE, {{"Input"}, {"Out"}}}, + {G_OP_TYPE_FUSION_CONV_BN_SCALE_RELU, {{"Input"}, {"Out"}}}, + {G_OP_TYPE_FUSION_POOL_BN, {{"X"}, {"Out"}}}, + {G_OP_TYPE_FUSION_ELEMENTWISE_ADD_RELU, {{"X", "Y"}, {"Out"}}}, + {G_OP_TYPE_REGION, {{"X"}, {"Out"}}}}; } // namespace paddle_mobile diff --git a/src/common/types.h b/src/common/types.h index 627b7efac6ef6bee6dc96295e63ee8f0f96b7932..d34c76710ad5fd40fb9d0c4ba67757f7a97558ff 100644 --- a/src/common/types.h +++ b/src/common/types.h @@ -102,6 +102,13 @@ extern const std::string G_OP_TYPE_DEPTHWISE_CONV; extern const std::string G_OP_TYPE_IM2SEQUENCE; extern const std::string G_OP_TYPE_DROPOUT; +extern const std::string G_OP_TYPE_FUSION_CONV_RELU; +extern const std::string G_OP_TYPE_FUSION_CONV_BN_SCALE; +extern const std::string G_OP_TYPE_FUSION_CONV_BN_SCALE_RELU; +extern const std::string G_OP_TYPE_FUSION_POOL_BN; +extern const std::string G_OP_TYPE_FUSION_ELEMENTWISE_ADD_RELU; +extern const std::string G_OP_TYPE_REGION; + extern std::unordered_map< std::string, std::pair, std::vector>> op_input_output_key;