insightface/SSH/rcnn/cython/gpu_nms.hpp

void _nms(int* keep_out, int* num_out, const float* boxes_host, int boxes_num,
          int boxes_dim, float nms_overlap_thresh, int device_id);