CMakeLists.txt 200 字节
nv_library(tensorrt_plugin
           SRCS trt_plugin.cc split_op_plugin.cu elementwise_op_plugin.cu prelu_op_plugin.cu
           avg_pool_op_plugin.cu
           DEPS enforce tensorrt_engine prelu)
反馈
建议
客服 返回
顶部