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

2 min read time

Applies to Linux

• include
• kernels
• device

Namespaces | Functions

thresholding_device.hpp File Reference

#include <hip/hip_runtime.h>
#include <core/wrappers/generic_tensor_wrapper.hpp>
#include "core/detail/type_traits.hpp"
#include "core/detail/casting.hpp"
#include "operator_types.h"

Go to the source code of this file.

Namespaces
	Kernels
	Kernels::Device

Functions
template<typename SrcWrapper , typename DstWrapper >
__global__ void	Kernels::Device::binary_generic (SrcWrapper input, DstWrapper output, roccv::GenericTensorWrapper< double > thresh, roccv::GenericTensorWrapper< double > maxVal, const int32_t maxBatchSize)
template<typename SrcWrapper , typename DstWrapper >
__global__ void	Kernels::Device::binary_inv_generic (SrcWrapper input, DstWrapper output, roccv::GenericTensorWrapper< double > thresh, roccv::GenericTensorWrapper< double > maxVal, const int32_t maxBatchSize)
template<typename SrcWrapper , typename DstWrapper >
__global__ void	Kernels::Device::tozero_generic (SrcWrapper input, DstWrapper output, roccv::GenericTensorWrapper< double > thresh, const int32_t maxBatchSize)
template<typename SrcWrapper , typename DstWrapper >
__global__ void	Kernels::Device::tozeroinv_generic (SrcWrapper input, DstWrapper output, roccv::GenericTensorWrapper< double > thresh, const int32_t maxBatchSize)
template<typename SrcWrapper , typename DstWrapper >
__global__ void	Kernels::Device::trunc_generic (SrcWrapper input, DstWrapper output, roccv::GenericTensorWrapper< double > thresh, const int32_t maxBatchSize)