From 254e74a840b6cdf9a7b3ef08575e049f6d3c7153 Mon Sep 17 00:00:00 2001 From: "Yao,kun" Date: Mon, 2 Jul 2018 11:29:23 +0800 Subject: [PATCH] update camke --- tools/op.cmake | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/tools/op.cmake b/tools/op.cmake index 2eabac925f..e35aa5248b 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() -- GitLab