NormalizeInInfer Struct Reference

NormalizeInInfer Struct Reference#

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

#include <element_wise_operation.hpp>

Public Member Functions

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

Public Attributes

double epsilon_

Static Public Attributes

static constexpr const char * name = "NormalizeInInfer"

Constructor & Destructor Documentation

◆ NormalizeInInfer()

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

Member Function Documentation

◆ operator()()

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

Member Data Documentation

◆ epsilon_

double ck::tensor_operation::element_wise::NormalizeInInfer::epsilon_

◆ name

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