Searched refs:numpy_nms_vmap (Results 1 – 1 of 1) sorted by relevance
426 def numpy_nms_vmap(info, in_dims, boxes, scores, iou_threshold): function429 numpy_nms.register_vmap(numpy_nms_vmap)