|
|
@ -227,6 +227,8 @@ op_library(softmax_op DEPS softmax)
|
|
|
|
op_library(sequence_softmax_op DEPS softmax)
|
|
|
|
op_library(sequence_softmax_op DEPS softmax)
|
|
|
|
if (WITH_GPU AND TENSORRT_FOUND)
|
|
|
|
if (WITH_GPU AND TENSORRT_FOUND)
|
|
|
|
op_library(tensorrt_engine_op DEPS tensorrt_engine)
|
|
|
|
op_library(tensorrt_engine_op DEPS tensorrt_engine)
|
|
|
|
|
|
|
|
else()
|
|
|
|
|
|
|
|
set(DEPS_OPS ${DEPS_OPS} tensorrt_engine_op)
|
|
|
|
endif()
|
|
|
|
endif()
|
|
|
|
op_library(sum_op DEPS selected_rows_functor)
|
|
|
|
op_library(sum_op DEPS selected_rows_functor)
|
|
|
|
op_library(sgd_op DEPS selected_rows_functor)
|
|
|
|
op_library(sgd_op DEPS selected_rows_functor)
|
|
|
|