op.cmake 12.5 KB
Newer Older
H
hanbuhe 已提交
1
set(FOUND_MATCH OFF)
2 3 4 5 6 7
set(CON -1)

message(STATUS "nets :${NET}")

list(FIND NET "googlenet" CON)
if (CON GREATER -1)
H
hanbuhe 已提交
8
  message("googlenet enabled")
D
dolphin8 已提交
9 10 11 12 13 14 15 16 17
  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)
Z
zhangyang 已提交
18
  set(FUSION_CONVADDRELU_OP ON)
H
hanbuhe 已提交
19 20 21 22

  set(FOUND_MATCH ON)
endif()

23 24
list(FIND NET "mobilenet" CON)
if (CON GREATER -1)
H
hanbuhe 已提交
25
  message("mobilenet enabled")
D
dolphin8 已提交
26 27 28 29
  set(CONV_OP ON)
  set(ELEMENTWISEADD_OP ON)
  set(RELU_OP ON)
  set(SOFTMAX_OP ON)
30
  set(MUL_OP ON)
D
dolphin8 已提交
31 32 33 34
  set(DEPTHWISECONV_OP ON)
  set(BATCHNORM_OP ON)
  set(POOL_OP ON)
  set(RESHAPE_OP ON)
W
wangliu 已提交
35
  set(FUSION_CONVADDBNRELU_OP ON)
36
  set(FUSION_CONVADDRELU_OP ON)
W
wangliu 已提交
37
  set(FUSION_CONVADD_OP ON)
H
hanbuhe 已提交
38 39 40 41 42

  set(FOUND_MATCH ON)
endif()


43 44
list(FIND NET "mobilenetssd" CON)
if (CON GREATER -1)
45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65
  message("mobilenetssd enabled")
  set(FUSION_CONVBNRELU_OP ON)
  set(FUSION_CONVBNRELU_OP ON)
  set(FUSION_DWCONVBNRELU_OP ON)
  set(FUSION_CONVADD_OP ON)
  set(MULTICLASSNMS_OP ON)
  set(SOFTMAX_OP ON)
  set(TRANSPOSE_OP ON)
    #feed
  set(PRIORBOX_OP ON)
  set(CONCAT_OP ON)
  set(BOXCODER_OP ON)
  set(RESHAPE_OP ON)
#fetch
  #total

  set(FOUND_MATCH ON)

endif()


66 67
list(FIND NET "yolo" CON)
if (CON GREATER -1)
H
hanbuhe 已提交
68
  message("yolo enabled")
D
dolphin8 已提交
69 70 71 72
  set(BATCHNORM_OP ON)
  set(CONV_OP ON)
  set(RELU_OP ON)
  set(ELEMENTWISEADD_OP ON)
H
hanbuhe 已提交
73 74 75 76

  set(FOUND_MATCH ON)
endif()

77 78
list(FIND NET "squeezenet" CON)
if (CON GREATER -1)
H
hanbuhe 已提交
79
  message("squeezenet enabled")
D
dolphin8 已提交
80 81 82 83 84 85 86
  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)
H
hanbuhe 已提交
87 88 89 90 91

  set(FOUND_MATCH ON)
endif()


92 93
list(FIND NET "resnet" CON)
if (CON GREATER -1)
H
hanbuhe 已提交
94 95
  message("resnet enabled")
  set(CONCAT_OP ON)
D
dolphin8 已提交
96
  set(CONV_OP ON)
H
hanbuhe 已提交
97
  set(RELU_OP ON)
D
dolphin8 已提交
98 99
  set(ELEMENTWISEADD_OP ON)
  set(POOL_OP ON)
qnqinan's avatar
qnqinan 已提交
100
  set(BATCHNORM_OP ON)
101
  set(FUSION_CONVBNADDRELU_OP ON)
qnqinan's avatar
qnqinan 已提交
102
  set(MUL_OP ON)
H
hanbuhe 已提交
103 104 105 106 107
  set(RESHAPE_OP ON)
  set(SOFTMAX_OP ON)
  set(FOUND_MATCH ON)
endif()

qnqinan's avatar
qnqinan 已提交
108
list(FIND NET "FPGA_NET_V1" CON)
109
if (CON GREATER -1)
qnqinan's avatar
qnqinan 已提交
110
  message("FPGA_NET_V1 enabled")
Z
zhangyang 已提交
111
  set(FUSION_CONVADDRELU_OP ON)
Z
zhangyang 已提交
112
  set(FUSION_ELEMENTWISEADDRELU_OP ON)
Z
zhangyang 已提交
113
  set(FUSION_FC_OP ON)
Z
zhangyang 已提交
114
  set(POOL_OP ON)
H
hanbuhe 已提交
115
  set(SOFTMAX_OP ON)
Z
zhangyang 已提交
116 117
  set(FUSION_CONVBNRELU_OP ON)
  set(FUSION_CONVBN_OP ON)
Z
zhangyang 已提交
118 119 120 121 122 123 124
  set(TANH_OP ON)
  set(ELEMENTWISEADD_OP ON)
  set(TRANSPOSE2_OP ON)
  set(FUSION_CONVADD_OP ON)
  set(SPLIT_OP ON)
  set(FUSION_DECONVADD_OP ON)
  set(FUSION_DECONVADDRELU_OP ON)
125 126 127 128 129

  set(RESHAPE_OP ON)
  set(FUSION_CONVADDBNRELU_OP ON)
  set(FUSION_CONVADDBN_OP ON)

xiebaiyuan's avatar
xiebaiyuan 已提交
130
  set(FOUND_MATCH ON)
H
hanbuhe 已提交
131 132
endif()

qnqinan's avatar
qnqinan 已提交
133 134 135
list(FIND NET "FPGA_NET_V2" CON)
if (CON GREATER -1)
  message("FPGA_NET_V2 enabled")
qnqinan's avatar
qnqinan 已提交
136
  set(FUSION_CONVADDRELU_OP ON)
qnqinan's avatar
qnqinan 已提交
137 138 139 140 141 142
  set(FUSION_ELEMENTWISEADDRELU_OP ON)
  set(FUSION_FC_OP ON)
  set(POOL_OP ON)
  set(SOFTMAX_OP ON)
  set(FUSION_CONVBNRELU_OP ON)
  set(FUSION_CONVBN_OP ON)
Z
zhangyang 已提交
143 144
  set(TANH_OP ON)
  set(ELEMENTWISEADD_OP ON)
qnqinan's avatar
qnqinan 已提交
145 146 147 148 149
  set(TRANSPOSE2_OP ON)
  set(FUSION_CONVADD_OP ON)
  set(SPLIT_OP ON)
  set(FUSION_DECONVADD_OP ON)
  set(FUSION_DECONVADDRELU_OP ON)
qnqinan's avatar
qnqinan 已提交
150 151 152
  set(FOUND_MATCH ON)
endif()

153 154
list(FIND NET "nlp" CON)
if (CON GREATER -1)
155 156 157 158 159 160 161 162 163 164 165 166
  message("nlp enabled")
  set(FUSION_FC_OP ON)
  set(LOOKUP_OP ON)
  set(GRU_OP ON)
  set(CRF_OP ON)
  set(CONCAT_OP ON)
  set(ELEMENTWISEADD_OP ON)


  set(FOUND_MATCH ON)
endif()

167 168
list(FIND NET "mobilenetfssd" CON)
if (CON GREATER -1)
xiebaiyuan's avatar
xiebaiyuan 已提交
169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188
  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()

189 190
list(FIND NET "genet" CON)
if (CON GREATER -1)
xiebaiyuan's avatar
xiebaiyuan 已提交
191 192 193 194 195 196 197 198 199 200 201 202 203
  message("genet enabled")
  set(FUSION_CONVADDPRELU_OP ON)
  set(FUSION_CONVADDADDPRELU_OP ON)
  set(FUSION_CONVADD_OP ON)
  set(CONV_TRANSPOSE_OP ON)
  set(FUSION_CONVADDRELU_OP ON)
  set(ELEMENTWISEADD_OP ON)
  set(PRELU_OP ON)
  set(POOL_OP ON)
  set(CONCAT_OP ON)

  set(FOUND_MATCH ON)
endif()
H
hanbuhe 已提交
204

205 206 207 208 209 210 211 212 213 214
list(FIND NET "super" CON)
if (CON GREATER -1)
  message("super enabled")
  set(FUSION_CONVADD_OP ON)
  set(FUSION_CONVADDRELU_OP ON)
  set(ELEMENTWISEADD_OP ON)

  set(FOUND_MATCH ON)
endif()

215

H
hanbuhe 已提交
216 217
if(NOT FOUND_MATCH)
  message("--default--")
Z
zhaojiaying01 已提交
218
  set(NORM_OP ON)
D
dolphin8 已提交
219
  set(BATCHNORM_OP ON)
L
liuruilong 已提交
220
  set(CONV_TRANSPOSE_OP ON)
D
dolphin8 已提交
221 222 223 224 225
  set(BOXCODER_OP ON)
  set(CONCAT_OP ON)
  set(CONV_OP ON)
  set(DEPTHWISECONV_OP ON)
  set(ELEMENTWISEADD_OP ON)
226 227
  set(ELEMENTWISESUB_OP ON)
  set(IM2SEQUENCE_OP ON)
L
lijiancheng0614 已提交
228
  set(FILL_CONSTANT_OP ON)
D
dolphin8 已提交
229
  set(FUSION_CONVADD_OP ON)
230
  set(FUSION_CONVADDPRELU_OP ON)
Z
zhangyang 已提交
231
  set(FUSION_CONVADDRELU_OP ON)
D
dolphin8 已提交
232 233 234 235
  set(FUSION_FC_OP ON)
  set(LRN_OP ON)
  set(MUL_OP ON)
  set(MULTICLASSNMS_OP ON)
L
lijiancheng0614 已提交
236
  set(POLYGONBOXTRANSFORM_OP ON)
D
dolphin8 已提交
237 238 239 240
  set(POOL_OP ON)
  set(PRIORBOX_OP ON)
  set(RELU_OP ON)
  set(RESHAPE_OP ON)
L
lijiancheng0614 已提交
241
  set(RESHAPE2_OP ON)
D
dolphin8 已提交
242 243 244
  set(SIGMOID_OP ON)
  set(SOFTMAX_OP ON)
  set(TRANSPOSE_OP ON)
L
lijiancheng0614 已提交
245
  set(TRANSPOSE2_OP ON)
E
eclipsess 已提交
246
  set(FUSION_CONVADDBNRELU_OP ON)
247
  set(FUSION_CONVADDADDPRELU_OP ON)
E
eclipsess 已提交
248
  set(FUSION_DWCONVBNRELU_OP ON)
249
  set(FUSION_CONVBNRELU_OP ON)
250
  set(FUSION_CONVBNADDRELU_OP ON)
T
Tian 已提交
251 252 253 254
  set(PRELU_OP ON)
  set(RESIZE_OP ON)
  set(SCALE_OP ON)
  set(SLICE_OP ON)
Y
Yao,kun 已提交
255 256
  set(DROPOUT_OP ON)
  set(IM2SEQUENCE_OP ON)
xiebaiyuan's avatar
xiebaiyuan 已提交
257 258
  set(LOOKUP_OP ON)
  set(GRU_OP ON)
Z
zhaojiaying01 已提交
259
  set(GRU_UNIT_OP ON)
xiebaiyuan's avatar
xiebaiyuan 已提交
260
  set(CRF_OP ON)
261 262 263 264
  set(BILINEAR_INTERP_OP ON)
  set(SPLIT_OP ON)
  set(FLATTEN_OP ON)
  set(SHAPE_OP ON)
E
debug  
eclipsess 已提交
265 266
  set(ELEMENTWISEMUL_OP ON)
  set(SUM_OP ON)
H
hjchen2 已提交
267 268
  set(TOP_K_OP ON)
  set(CAST_OP ON)
269 270
  set(QUANT_OP ON)
  set(DEQUANT_OP ON)
271
  set(FUSION_DEQUANT_BN_OP ON)
272 273 274
  set(FUSION_DEQUANT_ADD_BN_OP ON)
  set(FUSION_DEQUANT_BN_RELU_OP ON)
  set(FUSION_DEQUANT_ADD_BN_RELU_OP ON)
275 276
  set(FUSION_DEQUANT_ADD_BN_QUANT_OP ON)
  set(FUSION_DEQUANT_ADD_BN_RELU_QUANT_OP ON)
277 278
  set(SEQUENCE_EXPAND_OP ON)
  set(SEQUENCE_POOL_OP ON)
H
hjchen2 已提交
279
  set(SEQUENCE_SOFTMAX_OP ON)
280 281
  set(LOG_OP ON)
  set(TANH_OP ON)
282 283
  set(LOD_RESET_OP ON)
  set(LESS_THAN_OP ON)
284 285 286 287
  set(LOGICAL_AND_OP ON)
  set(LOGICAL_OR_OP ON)
  set(LOGICAL_NOT_OP ON)
  set(LOGICAL_XOR_OP ON)
288 289 290
  set(WHILE_OP ON)
  set(WRITE_TO_ARRAY_OP ON)
  set(READ_FROM_ARRAY_OP ON)
Z
zhaojiaying01 已提交
291 292
  set(IS_EMPTY_OP ON)
  set(INCREMENT_OP ON)
293 294 295
  set(ANCHOR_GENERATOR_OP ON)
  set(PROPOSAL_OP ON)
  set(PSROI_POOL_OP ON)
H
hjchen2 已提交
296
  set(ROI_PERSPECTIVE_OP ON)
H
update  
hjchen2 已提交
297
  set(BEAM_SEARCH_DECODE_OP ON)
H
hanbuhe 已提交
298 299
endif()

D
dolphin8 已提交
300 301 302 303 304 305
  # option(BATCHNORM_OP "" ON)
  # option(BOXCODER_OP "" ON)
  # option(CONCAT_OP "" ON)
  # option(CONV_OP "" ON)
  # option(DEPTHWISECONV_OP "" ON)
  # option(ELEMENTWISEADD_OP "" ON)
L
lijiancheng0614 已提交
306
  # option(FILL_CONSTANT_OP "" ON)
D
dolphin8 已提交
307
  # option(FUSION_CONVADD_OP "" ON)
Z
zhangyang 已提交
308
  # option(FUSION_CONVADDRELU_OP "" ON)
D
dolphin8 已提交
309 310 311 312
  # option(FUSION_FC_OP "" ON)
  # option(LRN_OP "" ON)
  # option(MUL_OP "" ON)
  # option(MULTICLASSNMS_OP "" ON)
L
lijiancheng0614 已提交
313
  # option(POLYGONBOXTRANSFORM_OP "" ON)
D
dolphin8 已提交
314 315 316 317
  # option(POOL_OP "" ON)
  # option(PRIORBOX_OP "" ON)
  # option(RELU_OP "" ON)
  # option(RESHAPE_OP "" ON)
L
lijiancheng0614 已提交
318
  # option(RESHAPE2_OP "" ON)
D
dolphin8 已提交
319 320 321
  # option(SIGMOID_OP "" ON)
  # option(SOFTMAX_OP "" ON)
  # option(TRANSPOSE_OP "" ON)
L
lijiancheng0614 已提交
322
  # option(TRANSPOSE2_OP "" ON)
H
hanbuhe 已提交
323
# endif ()
D
dolphin8 已提交
324

Z
zhaojiaying01 已提交
325 326 327
if (NORM_OP)
  add_definitions(-DNORM_OP)
endif()
D
dolphin8 已提交
328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344 345
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()
346 347 348
if (ELEMENTWISESUB_OP)
  add_definitions(-DELEMENTWISESUB_OP)
endif()
L
lijiancheng0614 已提交
349 350 351
if (FILL_CONSTANT_OP)
  add_definitions(-DFILL_CONSTANT_OP)
endif()
D
dolphin8 已提交
352 353 354
if (FUSION_CONVADD_OP)
  add_definitions(-DFUSION_CONVADD_OP)
endif()
Z
zhangyang 已提交
355 356
if (FUSION_CONVADDRELU_OP)
  add_definitions(-DFUSION_CONVADDRELU_OP)
D
dolphin8 已提交
357
endif()
358 359 360 361 362 363
if (FUSION_CONVADDPRELU_OP)
  add_definitions(-DFUSION_CONVADDPRELU_OP)
endif()
if (FUSION_CONVADDADDPRELU_OP)
  add_definitions(-DFUSION_CONVADDADDPRELU_OP)
endif()
D
dolphin8 已提交
364 365 366 367 368 369 370 371 372 373 374 375
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()
L
lijiancheng0614 已提交
376 377 378
if (POLYGONBOXTRANSFORM_OP)
  add_definitions(-DPOLYGONBOXTRANSFORM_OP)
endif()
D
dolphin8 已提交
379 380 381 382 383 384 385 386 387 388 389 390
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()
L
lijiancheng0614 已提交
391 392 393
if (RESHAPE2_OP)
  add_definitions(-DRESHAPE2_OP)
endif()
D
dolphin8 已提交
394 395 396 397 398 399 400 401 402
if (SIGMOID_OP)
  add_definitions(-DSIGMOID_OP)
endif()
if (SOFTMAX_OP)
  add_definitions(-DSOFTMAX_OP)
endif()
if (TRANSPOSE_OP)
  add_definitions(-DTRANSPOSE_OP)
endif()
L
lijiancheng0614 已提交
403 404 405
if (TRANSPOSE2_OP)
  add_definitions(-DTRANSPOSE2_OP)
endif()
E
eclipsess 已提交
406 407 408
if (FUSION_CONVADDBNRELU_OP)
  add_definitions(-DFUSION_CONVADDBNRELU_OP)
endif()
E
eclipsess 已提交
409 410 411
if (FUSION_DWCONVBNRELU_OP)
  add_definitions(-DFUSION_DWCONVBNRELU_OP)
endif()
412 413 414 415 416

if (FUSION_CONVBNRELU_OP)
  add_definitions(-DFUSION_CONVBNRELU_OP)
endif()

417 418 419 420
if (FUSION_CONVBNADDRELU_OP)
  add_definitions(-DFUSION_CONVBNADDRELU_OP)
endif()

T
Tian 已提交
421 422 423 424 425 426 427 428 429 430 431
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)
L
liuruilong 已提交
432
endif()
Y
Yao,kun 已提交
433 434 435
if (DROPOUT_OP)
  add_definitions(-DDROPOUT_OP)
endif()
Y
bugfix  
Yao,kun 已提交
436 437
if (IM2SEQUENCE_OP)
  add_definitions(-DIM2SEQUENCE_OP)
T
Tian 已提交
438
endif()
Z
zhangyang 已提交
439

Z
zhangyang 已提交
440 441
if (FUSION_CONVADDBN_OP)
  add_definitions(-DFUSION_CONVADDBN_OP)
Z
zhangyang 已提交
442
endif()
Z
zhangyang 已提交
443 444
if (FUSION_FCRELU_OP)
  add_definitions(-DFUSION_FCRELU_OP)
Z
zhangyang 已提交
445 446 447 448 449 450 451
endif()
if (FUSION_POOLBN_OP)
  add_definitions(-DFUSION_POOLBN_OP)
endif()
if (FUSION_ELEMENTWISEADDRELU_OP)
  add_definitions(-DFUSION_ELEMENTWISEADDRELU_OP)
endif()
Z
zhangyang 已提交
452 453
if (FUSION_CONVBN_OP)
  add_definitions(-DFUSION_CONVBN_OP)
Z
zhangyang 已提交
454 455
endif()

L
liuruilong 已提交
456
if (CONV_TRANSPOSE_OP)
H
hjchen2 已提交
457
  add_definitions(-DCONV_TRANSPOSE_OP)
L
liuruilong 已提交
458
endif()
xiebaiyuan's avatar
xiebaiyuan 已提交
459 460 461 462 463 464 465 466 467

if (LOOKUP_OP)
  add_definitions(-DLOOKUP_OP)
endif()

if (GRU_OP)
  add_definitions(-DGRU_OP)
endif()

Z
zhaojiaying01 已提交
468 469 470 471
if (GRU_UNIT_OP)
  add_definitions(-DGRU_UNIT_OP)
endif()

xiebaiyuan's avatar
xiebaiyuan 已提交
472 473 474
if (CRF_OP)
  add_definitions(-DCRF_OP)
endif()
475 476 477 478 479 480 481 482 483 484 485 486 487 488 489 490


if (FLATTEN_OP)
  add_definitions(-DFLATTEN_OP)
endif()

if (SPLIT_OP)
  add_definitions(-DSPLIT_OP)
endif()

if (BILINEAR_INTERP_OP)
  add_definitions(-DBILINEAR_INTERP_OP)
endif()

if (SHAPE_OP)
  add_definitions(-DSHAPE_OP)
H
hjchen2 已提交
491
endif()
E
debug  
eclipsess 已提交
492 493 494 495 496 497 498

if (ELEMENTWISEMUL_OP)
  add_definitions(-DELEMENTWISEMUL_OP)
endif()
if (SUM_OP)
  add_definitions(-DSUM_OP)
endif()
H
hjchen2 已提交
499 500 501 502 503 504
if (TOP_K_OP)
  add_definitions(-DTOP_K_OP)
endif()
if (CAST_OP)
  add_definitions(-DCAST_OP)
endif()
505 506 507 508 509 510
if (QUANT_OP)
  add_definitions(-DQUANT_OP)
endif()
if (DEQUANT_OP)
  add_definitions(-DDEQUANT_OP)
endif()
511 512 513
if (FUSION_DEQUANT_BN_OP)
  add_definitions(-DFUSION_DEQUANT_BN_OP)
endif()
514 515 516 517 518 519 520
if (FUSION_DEQUANT_ADD_BN_OP)
  add_definitions(-DFUSION_DEQUANT_ADD_BN_OP)
endif()
if (FUSION_DEQUANT_BN_RELU_OP)
  add_definitions(-DFUSION_DEQUANT_BN_RELU_OP)
endif()
if (FUSION_DEQUANT_ADD_BN_RELU_OP)
H
hjchen2 已提交
521 522
  add_definitions(-DFUSION_DEQUANT_ADD_BN_RELU_OP)
endif()
523 524 525 526 527 528
if (FUSION_DEQUANT_ADD_BN_QUANT_OP)
#  add_definitions(-DFUSION_DEQUANT_ADD_BN_QUANT_OP)
endif()
if (FUSION_DEQUANT_ADD_BN_RELU_QUANT_OP)
#  add_definitions(-DFUSION_DEQUANT_ADD_BN_RELU_QUANT_OP)
endif()
529 530 531 532 533 534
if (SEQUENCE_EXPAND_OP)
  add_definitions(-DSEQUENCE_EXPAND_OP)
endif()
if (SEQUENCE_POOL_OP)
  add_definitions(-DSEQUENCE_POOL_OP)
endif()
H
hjchen2 已提交
535 536 537
if (SEQUENCE_SOFTMAX_OP)
  add_definitions(-DSEQUENCE_SOFTMAX_OP)
endif()
538 539 540
if (LOG_OP)
  add_definitions(-DLOG_OP)
endif()
541 542 543 544 545 546
if (LOD_RESET_OP)
  add_definitions(-DLOD_RESET_OP)
endif()
if (LESS_THAN_OP)
  add_definitions(-DLESS_THAN_OP)
endif()
547 548 549 550 551 552 553 554 555 556 557 558
if (LOGICAL_AND_OP)
  add_definitions(-DLOGICAL_AND_OP)
endif()
if (LOGICAL_OR_OP)
  add_definitions(-DLOGICAL_OR_OP)
endif()
if (LOGICAL_NOT_OP)
  add_definitions(-DLOGICAL_NOT_OP)
endif()
if (LOGICAL_XOR_OP)
  add_definitions(-DLOGICAL_XOR_OP)
endif()
559

Z
zhangyang 已提交
560 561 562 563 564
if (TANH_OP)
  add_definitions(-DTANH_OP)
endif()
if (FUSION_DECONVRELU_OP)
  add_definitions(-DFUSION_DECONVRELU_OP)
qnqinan's avatar
qnqinan 已提交
565 566 567 568 569 570
endif()
if (FUSION_DECONVADD_OP)
  add_definitions(-DFUSION_DECONVADD_OP)
endif()
if (FUSION_DECONVADDRELU_OP)
  add_definitions(-DFUSION_DECONVADDRELU_OP)
H
hjchen2 已提交
571
endif()
572 573 574 575 576 577 578 579 580 581

if (WHILE_OP)
  add_definitions(-DWHILE_OP)
endif()
if (WRITE_TO_ARRAY_OP)
  add_definitions(-DWRITE_TO_ARRAY_OP)
endif()
if (READ_FROM_ARRAY_OP)
  add_definitions(-DREAD_FROM_ARRAY_OP)
endif()
Z
zhaojiaying01 已提交
582 583 584 585 586
if (IS_EMPTY_OP)
  add_definitions(-DIS_EMPTY_OP)
endif()
if (INCREMENT_OP)
  add_definitions(-DINCREMENT_OP)
587
endif()
588 589 590 591 592 593 594 595 596 597

if (ANCHOR_GENERATOR_OP)
  add_definitions(-DANCHOR_GENERATOR_OP)
endif()
if (PROPOSAL_OP)
  add_definitions(-DPROPOSAL_OP)
endif()
if (PSROI_POOL_OP)
  add_definitions(-DPSROI_POOL_OP)
endif()
H
hjchen2 已提交
598 599 600
if (ROI_PERSPECTIVE_OP)
  add_definitions(-DROI_PERSPECTIVE_OP)
endif()
H
update  
hjchen2 已提交
601 602 603
if (BEAM_SEARCH_DECODE_OP)
  add_definitions(-DBEAM_SEARCH_DECODE_OP)
endif()