|
|
|
@ -6,7 +6,7 @@ function(math_library TARGET)
|
|
|
|
|
# But it handle split GPU/CPU code and link some common library.
|
|
|
|
|
set(cc_srcs)
|
|
|
|
|
set(cu_srcs)
|
|
|
|
|
set(math_common_deps device_context)
|
|
|
|
|
set(math_common_deps device_context framework_proto)
|
|
|
|
|
set(multiValueArgs DEPS)
|
|
|
|
|
cmake_parse_arguments(math_library "${options}" "${oneValueArgs}"
|
|
|
|
|
"${multiValueArgs}" ${ARGN})
|
|
|
|
@ -35,7 +35,7 @@ math_library(depthwise_conv)
|
|
|
|
|
math_library(gru_compute DEPS activation_functions math_function)
|
|
|
|
|
math_library(im2col)
|
|
|
|
|
math_library(lstm_compute DEPS activation_functions)
|
|
|
|
|
math_library(math_function DEPS cblas framework_proto)
|
|
|
|
|
math_library(math_function DEPS cblas)
|
|
|
|
|
math_library(maxouting)
|
|
|
|
|
math_library(pooling)
|
|
|
|
|
math_library(selected_rows_functor DEPS selected_rows)
|
|
|
|
|