Normalize Struct Reference

Normalize Struct Reference#

Composable Kernel: ck::tensor_operation::element_wise::Normalize Struct Reference
Public Member Functions | Public Attributes | Static Public Attributes | List of all members
ck::tensor_operation::element_wise::Normalize Struct Reference

#include <element_wise_operation.hpp>

Public Member Functions

 Normalize (double epsilon=1e-4)
template<typename T1, typename T2, typename T3>
__host__ __device__ constexpr void operator() (T1 &y, const T1 &x, const T2 &mean, const T2 &mean_square, const T3 &gamma, const T3 &beta) const
template<>
__host__ __device__ constexpr void operator() (half_t &y, const half_t &x, const float &mean, const float &mean_square, const half_t &gamma, const half_t &beta) const
template<>
__host__ __device__ constexpr void operator() (float &y, const float &x, const float &mean, const float &mean_square, const float &gamma, const float &beta) const
template<>
__host__ __device__ constexpr void operator() (double &y, const double &x, const double &mean, const double &mean_square, const double &gamma, const double &beta) const

Public Attributes

double epsilon_

Static Public Attributes

static constexpr const char * name = "Normalize"

Constructor & Destructor Documentation

◆ Normalize()

ck::tensor_operation::element_wise::Normalize::Normalize ( double epsilon = 1e-4)
inline

Member Function Documentation

◆ operator()() [1/4]

template<>
__host__ __device__ constexpr void ck::tensor_operation::element_wise::Normalize::operator() ( double & y,
const double & x,
const double & mean,
const double & mean_square,
const double & gamma,
const double & beta ) const
inlineconstexpr

◆ operator()() [2/4]

template<>
__host__ __device__ constexpr void ck::tensor_operation::element_wise::Normalize::operator() ( float & y,
const float & x,
const float & mean,
const float & mean_square,
const float & gamma,
const float & beta ) const
inlineconstexpr

◆ operator()() [3/4]

template<>
__host__ __device__ constexpr void ck::tensor_operation::element_wise::Normalize::operator() ( half_t & y,
const half_t & x,
const float & mean,
const float & mean_square,
const half_t & gamma,
const half_t & beta ) const
inlineconstexpr

◆ operator()() [4/4]

template<typename T1, typename T2, typename T3>
__host__ __device__ constexpr void ck::tensor_operation::element_wise::Normalize::operator() ( T1 & y,
const T1 & x,
const T2 & mean,
const T2 & mean_square,
const T3 & gamma,
const T3 & beta ) const
constexpr

Member Data Documentation

◆ epsilon_

double ck::tensor_operation::element_wise::Normalize::epsilon_

◆ name

const char* ck::tensor_operation::element_wise::Normalize::name = "Normalize"
staticconstexpr
The documentation for this struct was generated from the following file: