Add_Mul2_Activation_Mul_Clamp< Activation > Struct Template Reference

• ck
• tensor_operation
• element_wise
• Add_Mul2_Activation_Mul_Clamp

Public Member Functions | Public Attributes | Static Public Attributes | List of all members

ck::tensor_operation::element_wise::Add_Mul2_Activation_Mul_Clamp< Activation > Struct Template Reference

#include <quantization_operation.hpp>

Public Member Functions
	Add_Mul2_Activation_Mul_Clamp (float scale_z_inv, Activation activationOp)
__host__ __device__ constexpr void	operator() (int8_t &y, const int32_t &x, const int32_t &bias, const float &scaleAcc) const
__host__ __device__ constexpr void	operator() (int32_t &y, const int32_t &x, const int32_t &bias, const float &scaleAcc) const

Public Attributes
float	scale_z_inv_
Activation	activationOp_

Static Public Attributes
static constexpr const char *	name = "Add_Mul2_Activation_Mul_Clamp"

Constructor & Destructor Documentation

◆ Add_Mul2_Activation_Mul_Clamp()

template<typename Activation>

ck::tensor_operation::element_wise::Add_Mul2_Activation_Mul_Clamp< Activation >::Add_Mul2_Activation_Mul_Clamp ( float scale_z_inv, Activation activationOp )

inline

Member Function Documentation

◆ operator()() [1/2]

template<typename Activation>

__host__ __device__ constexpr void ck::tensor_operation::element_wise::Add_Mul2_Activation_Mul_Clamp< Activation >::operator() ( int32_t & y, const int32_t & x, const int32_t & bias, const float & scaleAcc ) const

inlineconstexpr

◆ operator()() [2/2]

template<typename Activation>

__host__ __device__ constexpr void ck::tensor_operation::element_wise::Add_Mul2_Activation_Mul_Clamp< Activation >::operator() ( int8_t & y, const int32_t & x, const int32_t & bias, const float & scaleAcc ) const

inlineconstexpr

Member Data Documentation

◆ activationOp_

template<typename Activation>

Activation ck::tensor_operation::element_wise::Add_Mul2_Activation_Mul_Clamp< Activation >::activationOp_

◆ name

template<typename Activation>

const char* ck::tensor_operation::element_wise::Add_Mul2_Activation_Mul_Clamp< Activation >::name = "Add_Mul2_Activation_Mul_Clamp"

staticconstexpr

◆ scale_z_inv_

template<typename Activation>

float ck::tensor_operation::element_wise::Add_Mul2_Activation_Mul_Clamp< Activation >::scale_z_inv_

---

The documentation for this struct was generated from the following file:

• quantization_operation.hpp