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

3 4 5 6 7 8 9
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})
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 已提交
10 11
cc_library(activation_ops_lite SRCS activation_ops.cc DEPS ${op_DEPS})
cc_library(elementwise_ops_lite SRCS elementwise_ops.cc DEPS ${op_DEPS})
S
superjomn 已提交
12

13
cc_library(op_params_lite SRCS op_params.cc DEPS ${tensor_lite})
S
update  
superjomn 已提交
14
set(ops_lite
S
superjomn 已提交
15
        fc_op_lite
S
superjomn 已提交
16 17 18
        relu_op_lite
        mul_op_lite
        scale_op_lite
S
Superjomn 已提交
19
        feed_op_lite
20
        fetch_op_lite
S
Superjomn 已提交
21
        io_copy_op_lite
S
update  
superjomn 已提交
22
        PARENT_SCOPE)
S
superjomn 已提交
23

S
update  
superjomn 已提交
24
lite_cc_test(test_fc_op_lite SRCS fc_op_test.cc DEPS fc_op_lite fc_compute_host)