BiasNormalizeInInferClamp Struct Reference

BiasNormalizeInInferClamp Struct Reference#

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

#include <element_wise_operation.hpp>

Public Member Functions

 BiasNormalizeInInferClamp (float floor=0.f, float ceil=NumericLimits< float >::Max(), float epsilon=1e-4)
template<typename T>
__host__ __device__ constexpr void operator() (T &y, const T &x, const T &bias, const T &mean, const T &variance, const T &gamma, const T &beta) const
template<>
__host__ __device__ constexpr void operator() (float &y, const float &x, const float &bias, const float &mean, const float &variance, const float &gamma, const float &beta) const

Public Attributes

Clamp clamp_
float epsilon_

Static Public Attributes

static constexpr const char * name = "BiasNormalizeInInferClamp"

Constructor & Destructor Documentation

◆ BiasNormalizeInInferClamp()

ck::tensor_operation::element_wise::BiasNormalizeInInferClamp::BiasNormalizeInInferClamp ( float floor = 0.f,
float ceil = NumericLimits<float>::Max(),
float epsilon = 1e-4 )
inline

Member Function Documentation

◆ operator()() [1/2]

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

◆ operator()() [2/2]

template<typename T>
__host__ __device__ constexpr void ck::tensor_operation::element_wise::BiasNormalizeInInferClamp::operator() ( T & y,
const T & x,
const T & bias,
const T & mean,
const T & variance,
const T & gamma,
const T & beta ) const
inlineconstexpr

Member Data Documentation

◆ clamp_

Clamp ck::tensor_operation::element_wise::BiasNormalizeInInferClamp::clamp_

◆ epsilon_

float ck::tensor_operation::element_wise::BiasNormalizeInInferClamp::epsilon_

◆ name

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