CMakeLists.txt 3.6 KB
Newer Older
T
tensor-tang 已提交
1 2
set(op_DEPS ${tensor_lite} op_lite op_params_lite)

Z
Zhen Wang 已提交
3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28
lite_cc_library(conv_op_lite SRCS conv_op.cc DEPS ${op_DEPS})
lite_cc_library(pool_op_lite SRCS pool_op.cc DEPS ${op_DEPS})
lite_cc_library(fc_op_lite SRCS fc_op.cc DEPS ${op_DEPS})
lite_cc_library(relu_op_lite SRCS relu_op.cc DEPS ${op_DEPS})
lite_cc_library(mul_op_lite SRCS mul_op.cc DEPS ${op_DEPS})
lite_cc_library(scale_op_lite SRCS scale_op.cc DEPS ${op_DEPS})
lite_cc_library(softmax_op_lite SRCS softmax_op.cc DEPS ${op_DEPS})
lite_cc_library(reshape_op_lite SRCS reshape_op.cc DEPS ${op_DEPS} )
lite_cc_library(batch_norm_op_lite SRCS batch_norm_op.cc DEPS ${op_DEPS})
lite_cc_library(feed_op_lite SRCS feed_op.cc DEPS ${op_DEPS})
lite_cc_library(fetch_op_lite SRCS fetch_op.cc DEPS ${op_DEPS})
lite_cc_library(io_copy_op_lite SRCS io_copy_op.cc DEPS ${op_DEPS})
lite_cc_library(activation_ops_lite SRCS activation_ops.cc DEPS ${op_DEPS})
lite_cc_library(elementwise_ops_lite SRCS elementwise_ops.cc DEPS ${op_DEPS})
lite_cc_library(fusion_elementwise_activation_ops_lite SRCS fusion_elementwise_activation_ops.cc DEPS elementwise_ops_lite ${op_DEPS})
lite_cc_library(mean_op_lite SRCS mean_op.cc DEPS ${op_DEPS})
lite_cc_library(fill_constant_op_lite SRCS fill_constant_op.cc DEPS ${op_DEPS})
#lite_cc_library(sgd_op_lite SRCS sgd_op.cc DEPS ${op_DEPS})
lite_cc_library(op_params_lite SRCS op_params.cc DEPS ${tensor_lite} any_lite framework_proto_lite)
lite_cc_library(dropout_op_lite SRCS dropout_op.cc DEPS ${op_DEPS})
lite_cc_library(concat_op_lite SRCS concat_op.cc DEPS ${op_DEPS})
lite_cc_library(calib_op_lite SRCS calib_op.cc DEPS ${op_DEPS})
lite_cc_library(split_op_lite SRCS split_op.cc DEPS ${op_DEPS})
lite_cc_library(transpose_op_lite SRCS transpose_op.cc DEPS ${op_DEPS})
lite_cc_library(fake_quant SRCS fake_quantize_moving_avg_max_abs.cc DEPS ${op_DEPS})
lite_cc_library(fake_dequant SRCS fake_dequantize_max_abs.cc DEPS ${op_DEPS})
T
tensor-tang 已提交
29 30 31

set(ops_lite
        conv_op_lite
32
        pool_op_lite
T
tensor-tang 已提交
33 34 35 36 37 38
        fc_op_lite
        relu_op_lite
        mul_op_lite
        scale_op_lite
        softmax_op_lite
        reshape_op_lite
39
        batch_norm_op_lite
T
tensor-tang 已提交
40 41 42 43
        feed_op_lite
        fetch_op_lite
        io_copy_op_lite
        elementwise_ops_lite
44
        fusion_elementwise_activation_ops_lite
T
tensor-tang 已提交
45 46 47 48 49
        mean_op_lite
        fill_constant_op_lite
        activation_ops_lite
        dropout_op_lite
        concat_op_lite
S
Shixiaowei02 已提交
50
        calib_op_lite
Z
zhupy 已提交
51
        split_op_lite
52
        transpose_op_lite
N
nhzlx 已提交
53 54
        fake_quant
        fake_dequant
T
tensor-tang 已提交
55 56 57 58 59
        PARENT_SCOPE)

lite_cc_test(test_fc_op_lite SRCS fc_op_test.cc 
             DEPS fc_op_lite memory_lite 
             X86_DEPS fc_compute_x86
60 61 62 63
             ARM_DEPS fc_compute_arm)
lite_cc_test(test_pool_op_lite SRCS pool_op_test.cc
             DEPS pool_op_lite memory_lite
             ARM_DEPS pool_compute_arm)
T
tensor-tang 已提交
64 65 66
lite_cc_test(test_scale_op_lite SRCS scale_op_test.cc DEPS scale_op_lite memory_lite)
lite_cc_test(test_softmax_op_lite SRCS softmax_op_test.cc DEPS softmax_op_lite memory_lite)
lite_cc_test(test_reshape_op_lite SRCS reshape_op_test.cc DEPS reshape_op_lite memory_lite)
67
lite_cc_test(test_batch_norm_op_lite SRCS batch_norm_op_test.cc DEPS batch_norm_op_lite memory_lite)
T
tensor-tang 已提交
68
lite_cc_test(test_concat_op_lite SRCS concat_op_test.cc DEPS concat_op_lite memory_lite)
S
Shixiaowei02 已提交
69
lite_cc_test(test_calib_op_lite SRCS calib_op_test.cc DEPS calib_op_lite memory_lite ARM_DEPS calib_compute_arm)
70 71 72
lite_cc_test(test_fusion_elementwise_activation_ops_lite 
             SRCS fusion_elementwise_activation_ops_test.cc 
             DEPS fusion_elementwise_activation_ops_lite memory_lite)
73
lite_cc_test(test_transpose_op_lite SRCS transpose_op_test.cc DEPS transpose_op_lite memory_lite)