q.yao 3fa94f4424

[Feature] better tensorrt cpp code (#11) ...

* better tensorrt cpp code

* fix end of file

2021-07-12 10:01:52 +08:00

..

allClassNMS.cu

[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_kernel.cpp

[Feature] better tensorrt cpp code (#11)

2021-07-12 10:01:52 +08:00

trt_batched_nms_kernel.hpp

[Feature] better tensorrt cpp code (#11)

2021-07-12 10:01:52 +08:00

trt_batched_nms.cpp

[Feature] better tensorrt cpp code (#11)

2021-07-12 10:01:52 +08:00

trt_batched_nms.hpp

[Feature] better tensorrt cpp code (#11)

2021-07-12 10:01:52 +08:00