CMakeLists.txt 160 字节
Newer Older
1 2 3
nv_library(tensorrt_plugin
           SRCS trt_plugin.cc split_op_plugin.cu elementwise_op_plugin.cu prelu_op_plugin.cu
           DEPS enforce device_context)