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

1 min read time

Applies to Linux

• include
• kernels
• device

Namespaces | Functions

normalize_device.hpp File Reference

Contains the device kernel implementation for the Normalize operation. More...

#include <hip/hip_runtime.h>
#include "core/detail/casting.hpp"
#include "core/detail/math/vectorized_type_math.hpp"
#include "core/detail/type_traits.hpp"
#include "core/detail/vector_utils.hpp"

Go to the source code of this file.

Namespaces
	Kernels
	Kernels::Device

Functions
template<bool ScaleStddev, typename SrcWrapper , typename DstWrapper , typename ScaleWrapper , typename BaseWrapper >
__global__ void	Kernels::Device::normalize (SrcWrapper input, BaseWrapper base, ScaleWrapper scale, DstWrapper output, float globalScale, float shift, float epsilon)

Detailed Description

Contains the device kernel implementation for the Normalize operation.