mmdeploy/backend_ops/tensorrt/batched_nms
grimoire 8fd4ef4d47 remove unused flag, fix lint 2021-06-29 10:23:52 +08:00
..
allClassNMS.cu [Feature] Add TensorRT batched NMS support (#3) 2021-06-25 19:31:16 +08:00
batchedNMSInference.cpp [Feature] Add TensorRT batched NMS support (#3) 2021-06-25 19:31:16 +08:00
cub_helper.h [Feature] Add TensorRT batched NMS support (#3) 2021-06-25 19:31:16 +08:00
gatherNMSOutputs.cu [Feature] Add TensorRT batched NMS support (#3) 2021-06-25 19:31:16 +08:00
kernel.cu remove unused flag, fix lint 2021-06-29 10:23:52 +08:00
kernel.h remove unused flag, fix lint 2021-06-29 10:23:52 +08:00
permuteData.cu [Feature] Add TensorRT batched NMS support (#3) 2021-06-25 19:31:16 +08:00
sortScoresPerClass.cu [Feature] Add TensorRT batched NMS support (#3) 2021-06-25 19:31:16 +08:00
sortScoresPerImage.cu [Feature] Add TensorRT batched NMS support (#3) 2021-06-25 19:31:16 +08:00
trt_batched_nms.cpp add CI 2021-06-29 10:10:15 +08:00
trt_batched_nms.hpp [Feature] Add TensorRT batched NMS support (#3) 2021-06-25 19:31:16 +08:00