IBM_AI/vi-lib/vi_od_frcnn/nms/gpu_nms.hpp

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