/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-roccv/checkouts/latest/include/kernels/device/non_max_suppression_device.hpp File Reference

1 min read time

Applies to Linux

• include
• kernels
• device

Namespaces | Functions

non_max_suppression_device.hpp File Reference

#include <hip/hip_runtime.h>
#include "core/tensor.hpp"
#include "core/wrappers/generic_tensor_wrapper.hpp"
#include "kernels/common/non_max_suppression_helpers.hpp"

Go to the source code of this file.

Namespaces
	Kernels
	Kernels::Device

Functions
__global__ void	Kernels::Device::non_maximum_suppression (roccv::GenericTensorWrapper< short4 > input, roccv::GenericTensorWrapper< uint8_t > output, roccv::GenericTensorWrapper< float > scores, int numBoxes, float scoresThreshold, float iouThreshold)