src/pixel3dmm/preprocessing/PIPNet/FaceBoxesV2/utils/nms/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);