diff --git a/tools/op.cmake b/tools/op.cmake index 2eabac925f6021448243b3668c22cbcaebe2f1d9..e35aa5248b0f4b1844bc71e88a23ac54ac4eaf31 100644 --- a/tools/op.cmake +++ b/tools/op.cmake @@ -64,6 +64,8 @@ else () set(SOFTMAX_OP ON) set(TRANSPOSE_OP ON) set(FUSION_CONVADD_RELU_OP ON) + set(DROPOUT_OP ON) + set(IM2SEQUENCE_OP ON) # option(BATCHNORM_OP "" ON) # option(BOXCODER_OP "" ON) # option(CONCAT_OP "" ON) @@ -145,4 +147,10 @@ if (TRANSPOSE_OP) endif() if (FUSION_CONVADD_RELU_OP) add_definitions(-DFUSION_CONVADD_RELU_OP) -endif() \ No newline at end of file +endif() +if (DROPOUT_OP) + add_definitions(-DDROPOUT_OP) +endif() +if (IM2SQUENCE_OP) + add_definitions(-DIM2SQUENCE_OP) +endif()