CMakeLists.txt 1.6 KB
Newer Older
1
set(op_DEPS ${tensor_lite} op_lite op_params_lite)
S
update  
superjomn 已提交
2

3 4 5 6
cc_library(fc_op_lite SRCS fc_op.cc DEPS ${op_DEPS})
cc_library(relu_op_lite SRCS relu_op.cc DEPS ${op_DEPS})
cc_library(mul_op_lite SRCS mul_op.cc DEPS ${op_DEPS})
cc_library(scale_op_lite SRCS scale_op.cc DEPS ${op_DEPS})
7
cc_library(softmax_op_lite SRCS softmax_op.cc DEPS ${op_DEPS})
8 9 10
cc_library(feed_op_lite SRCS feed_op.cc DEPS ${op_DEPS})
cc_library(fetch_op_lite SRCS fetch_op.cc DEPS ${op_DEPS})
cc_library(io_copy_op_lite SRCS io_copy_op.cc DEPS ${op_DEPS})
Y
Yan Chunwei 已提交
11 12
cc_library(activation_ops_lite SRCS activation_ops.cc DEPS ${op_DEPS})
cc_library(elementwise_ops_lite SRCS elementwise_ops.cc DEPS ${op_DEPS})
L
liuwei1031 已提交
13 14
cc_library(mean_op_lite SRCS mean_op.cc DEPS ${op_DEPS})
cc_library(fill_constant_op_lite SRCS fill_constant_op.cc DEPS ${op_DEPS})
L
liuwei1031 已提交
15
#cc_library(sgd_op_lite SRCS sgd_op.cc DEPS ${op_DEPS})
Y
Yan Chunwei 已提交
16
cc_library(op_params_lite SRCS op_params.cc DEPS ${tensor_lite} any_lite framework_proto_lite)
17 18
cc_library(dropout_op_lite SRCS dropout_op.cc DEPS ${op_DEPS})

S
update  
superjomn 已提交
19
set(ops_lite
S
superjomn 已提交
20
        fc_op_lite
S
superjomn 已提交
21 22 23
        relu_op_lite
        mul_op_lite
        scale_op_lite
24
        softmax_op_lite
S
Superjomn 已提交
25
        feed_op_lite
26
        fetch_op_lite
S
Superjomn 已提交
27
        io_copy_op_lite
L
liuwei1031 已提交
28 29 30
        elementwise_ops_lite
        mean_op_lite
        fill_constant_op_lite
31 32
        activation_ops_lite
        dropout_op_lite
S
update  
superjomn 已提交
33
        PARENT_SCOPE)
S
superjomn 已提交
34

35 36 37 38
lite_cc_test(test_fc_op_lite SRCS fc_op_test.cc 
             DEPS fc_op_lite memory_lite 
             X86_DEPS fc_compute_x86
	         ARM_DEPS fc_compute_arm)
39
lite_cc_test(test_softmax_op_lite SRCS softmax_op_test.cc DEPS softmax_op_lite memory_lite)