• q.yao's avatar
    NMS with CUDA only (#1824) · 74031cc5
    q.yao authored
    * add gather_keep_from_mask_parallize
    
    * remove unused cache
    
    * move syncthread
    
    * remove unused comment
    
    * add more comments, rename the kernel and variable
    74031cc5
nms_cuda_kernel.cuh 3.92 KB