torchvision_nms_kernel.hpp File Reference

Declaration for the vendored torchvision CPU NMS kernel. More...

#include <ATen/ATen.h>

Include dependency graph for torchvision_nms_kernel.hpp:

Go to the source code of this file.

Namespaces
namespace	reusex
namespace	reusex::vision
namespace	reusex::vision::torchvision

Functions
at::Tensor	reusex::vision::torchvision::nms_kernel (const at::Tensor &dets, const at::Tensor &scores, double iou_threshold)
	torchvision's reference CPU non-maximum-suppression kernel.

Detailed Description

Declaration for the vendored torchvision CPU NMS kernel.

See src/vision/third_party/torchvision_nms_kernel.cpp for provenance and the list of adaptations made to the upstream source.

Callers inside ReUseX should not use this directly — go through reusex::vision::nms() in vision/nms.hpp, which owns the ReUseX-facing preconditions and API contract.

Definition in file torchvision_nms_kernel.hpp.