set(FOUND_MATCH OFF) if ("googlenet" IN_LIST NET) message("googlenet enabled") set(CONCAT_OP ON) set(CONV_OP ON) set(LRN_OP ON) set(MUL_OP ON) set(ELEMENTWISEADD_OP ON) set(FUSION_FC_OP ON) set(POOL_OP ON) set(RELU_OP ON) set(FUSION_CONVADD_OP ON) set(FUSION_CONVADDRELU_OP ON) set(FOUND_MATCH ON) endif() if ("mobilenet" IN_LIST NET) message("mobilenet enabled") set(CONV_OP ON) set(ELEMENTWISEADD_OP ON) set(RELU_OP ON) set(SOFTMAX_OP ON) set(MUL_OP ON) set(DEPTHWISECONV_OP ON) set(BATCHNORM_OP ON) set(POOL_OP ON) set(RESHAPE_OP ON) set(FUSION_CONVADDBNRELU_OP ON) set(FUSION_CONVADD_OP ON) set(FOUND_MATCH ON) endif() if ("yolo" IN_LIST NET) message("yolo enabled") set(BATCHNORM_OP ON) set(CONV_OP ON) set(RELU_OP ON) set(ELEMENTWISEADD_OP ON) set(FOUND_MATCH ON) endif() if ("squeezenet" IN_LIST NET) message("squeezenet enabled") set(CONCAT_OP ON) set(CONV_OP ON) set(RELU_OP ON) set(ELEMENTWISEADD_OP ON) set(POOL_OP ON) set(RESHAPE_OP ON) set(SOFTMAX_OP ON) set(FOUND_MATCH ON) endif() if ("resnet" IN_LIST NET) message("resnet enabled") set(CONCAT_OP ON) set(CONV_OP ON) set(RELU_OP ON) set(ELEMENTWISEADD_OP ON) set(POOL_OP ON) set(RESHAPE_OP ON) set(SOFTMAX_OP ON) set(FOUND_MATCH ON) endif() if ("FPGAnets" IN_LIST NET) message("FPGAnets enabled") set(FUSION_CONVADDRELU_OP ON) set(FUSION_CONVADDBNRELU_OP ON) set(FUSION_CONVADDBN_OP ON) set(FUSION_ELEMENTWISEADDRELU_OP ON) set(FUSION_FC_OP ON) set(FUSION_FCRELU_OP ON) set(POOL_OP ON) set(CONCAT_OP ON) set(SOFTMAX_OP ON) set(DROPOUT_OP ON) set(FOUND_MATCH ON) endif() if(NOT FOUND_MATCH) message("--default--") set(BATCHNORM_OP ON) set(CONV_TRANSPOSE_OP ON) set(BOXCODER_OP ON) set(CONCAT_OP ON) set(CONV_OP ON) set(DEPTHWISECONV_OP ON) set(ELEMENTWISEADD_OP ON) set(FUSION_CONVADD_OP ON) set(FUSION_CONVADDRELU_OP ON) set(FUSION_FC_OP ON) set(LRN_OP ON) set(MUL_OP ON) set(MULTICLASSNMS_OP ON) set(POOL_OP ON) set(PRIORBOX_OP ON) set(RELU_OP ON) set(RESHAPE_OP ON) set(SIGMOID_OP ON) set(SOFTMAX_OP ON) set(TRANSPOSE_OP ON) set(FUSION_CONVADDBNRELU_OP ON) set(FUSION_DWCONVBNRELU_OP ON) set(FUSION_CONVBNRELU_OP ON) set(PRELU_OP ON) set(RESIZE_OP ON) set(SCALE_OP ON) set(SLICE_OP ON) set(DROPOUT_OP ON) set(IM2SEQUENCE_OP ON) endif() # option(BATCHNORM_OP "" ON) # option(BOXCODER_OP "" ON) # option(CONCAT_OP "" ON) # option(CONV_OP "" ON) # option(DEPTHWISECONV_OP "" ON) # option(ELEMENTWISEADD_OP "" ON) # option(FUSION_CONVADD_OP "" ON) # option(FUSION_CONVADDRELU_OP "" ON) # option(FUSION_FC_OP "" ON) # option(LRN_OP "" ON) # option(MUL_OP "" ON) # option(MULTICLASSNMS_OP "" ON) # option(POOL_OP "" ON) # option(PRIORBOX_OP "" ON) # option(RELU_OP "" ON) # option(RESHAPE_OP "" ON) # option(SIGMOID_OP "" ON) # option(SOFTMAX_OP "" ON) # option(TRANSPOSE_OP "" ON) # endif () if (BATCHNORM_OP) add_definitions(-DBATCHNORM_OP) endif() if (BOXCODER_OP) add_definitions(-DBOXCODER_OP) endif() if (CONCAT_OP) add_definitions(-DCONCAT_OP) endif() if (CONV_OP) add_definitions(-DCONV_OP) endif() if (DEPTHWISECONV_OP) add_definitions(-DDEPTHWISECONV_OP) endif() if (ELEMENTWISEADD_OP) add_definitions(-DELEMENTWISEADD_OP) endif() if (FUSION_CONVADD_OP) add_definitions(-DFUSION_CONVADD_OP) endif() if (FUSION_CONVADDRELU_OP) add_definitions(-DFUSION_CONVADDRELU_OP) endif() if (FUSION_FC_OP) add_definitions(-DFUSION_FC_OP) endif() if (LRN_OP) add_definitions(-DLRN_OP) endif() if (MUL_OP) add_definitions(-DMUL_OP) endif() if (MULTICLASSNMS_OP) add_definitions(-DMULTICLASSNMS_OP) endif() if (POOL_OP) add_definitions(-DPOOL_OP) endif() if (PRIORBOX_OP) add_definitions(-DPRIORBOX_OP) endif() if (RELU_OP) add_definitions(-DRELU_OP) endif() if (RESHAPE_OP) add_definitions(-DRESHAPE_OP) endif() if (SIGMOID_OP) add_definitions(-DSIGMOID_OP) endif() if (SOFTMAX_OP) add_definitions(-DSOFTMAX_OP) endif() if (TRANSPOSE_OP) add_definitions(-DTRANSPOSE_OP) endif() if (FUSION_CONVADDBNRELU_OP) add_definitions(-DFUSION_CONVADDBNRELU_OP) endif() if (FUSION_DWCONVBNRELU_OP) add_definitions(-DFUSION_DWCONVBNRELU_OP) endif() if (FUSION_CONVBNRELU_OP) add_definitions(-DFUSION_CONVBNRELU_OP) endif() if (PRELU_OP) add_definitions(-DPRELU_OP) endif() if (RESIZE_OP) add_definitions(-DRESIZE_OP) endif() if (SCALE_OP) add_definitions(-DSCALE_OP) endif() if (SLICE_OP) add_definitions(-DSLICE_OP) endif() if (DROPOUT_OP) add_definitions(-DDROPOUT_OP) endif() if (IM2SEQUENCE_OP) add_definitions(-DIM2SEQUENCE_OP) endif() if (FUSION_CONVADDBN_OP) add_definitions(-DFUSION_CONVADDBN_OP) endif() if (FUSION_FCRELU_OP) add_definitions(-DFUSION_FCRELU_OP) endif() if (FUSION_POOLBN_OP) add_definitions(-DFUSION_POOLBN_OP) endif() if (FUSION_ELEMENTWISEADDRELU_OP) add_definitions(-DFUSION_ELEMENTWISEADDRELU_OP) endif() if (REGION_OP) add_definitions(-DREGION_OP) endif() if (CONV_TRANSPOSE_OP) add_definitions(-DCONV_TRANSPOSE) endif()