提交 30eccaa3 编写于 作者: xiebaiyuan's avatar xiebaiyuan

add single build for mobilenetfssd

上级 9c44f9ae
......@@ -13,6 +13,11 @@ See the License for the specific language governing permissions and
limitations under the License. */
#ifdef FLATTEN_OP
#ifndef RESHAPE_OP
#define RESHAPE_OP
#endif
#pragma once
#include <operators/kernel/reshape_kernel.h>
......
......@@ -51,6 +51,10 @@ elseif("nlp" IN_LIST NET)
# gen test
ADD_EXECUTABLE(test-gru-op operators/test_gru_op.cpp test_helper.h test_include.h)
target_link_libraries(test-gru-op paddle-mobile)
elseif("mobilenetfssd" IN_LIST NET)
# gen test
ADD_EXECUTABLE(test-fssd net/test_mobilenet_025_fssd.cpp test_helper.h test_include.h)
target_link_libraries(test-fssd paddle-mobile)
else ()
# gen test
......
#!/usr/bin/env bash
NETS=""
declare -a supportedNets=("googlenet" "mobilenet" "yolo" "squeezenet" "resnet" "mobilenetssd" "nlp")
declare -a supportedNets=("googlenet" "mobilenet" "yolo" "squeezenet" "resnet" "mobilenetssd" "nlp" "mobilenetfssd")
build_for_mac() {
if [ ! `which brew` ]; then
......
......@@ -127,6 +127,27 @@ if ("nlp" IN_LIST NET)
set(FOUND_MATCH ON)
endif()
if ("mobilenetfssd" IN_LIST NET)
message("mobilenetfssd enabled")
set(FUSION_CONVADDRELU_OP ON)
set(FUSION_CONVADDBNRELU_OP ON)
set(FUSION_CONVADD_OP ON)
set(SOFTMAX_OP ON)
set(RESHAPE_OP ON)
set(BILINEAR_INTERP_OP ON)
set(TRANSPOSE_OP ON)
set(CONCAT_OP ON)
set(PRIORBOX_OP ON)
set(BATCHNORM_OP ON)
set(BOXCODER_OP ON)
set(MULTICLASSNMS_OP ON)
set(FLATTEN_OP ON)
set(SPLIT_OP ON)
set(SHAPE_OP ON)
set(FOUND_MATCH ON)
endif()
if(NOT FOUND_MATCH)
message("--default--")
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册