Add_Mul_Activation_Mul_Clamp< Activation > Struct Template Reference

Add_Mul_Activation_Mul_Clamp&lt; Activation &gt; Struct Template Reference#

Composable Kernel: ck::tensor_operation::element_wise::Add_Mul_Activation_Mul_Clamp< Activation > Struct Template Reference
Public Member Functions | Public Attributes | Static Public Attributes | List of all members
ck::tensor_operation::element_wise::Add_Mul_Activation_Mul_Clamp< Activation > Struct Template Reference

#include <quantization_operation.hpp>

Public Member Functions

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

Public Attributes

float scale_z_inv_
float scaleAcc_
Activation activationOp_

Static Public Attributes

static constexpr const char * name = "Add_Mul_Activation_Mul_Clamp"

Constructor & Destructor Documentation

◆ Add_Mul_Activation_Mul_Clamp()

template<typename Activation>
ck::tensor_operation::element_wise::Add_Mul_Activation_Mul_Clamp< Activation >::Add_Mul_Activation_Mul_Clamp ( float scale_z_inv,
float scaleAcc,
Activation activationOp )
inline

Member Function Documentation

◆ operator()() [1/2]

template<typename Activation>
__host__ __device__ constexpr void ck::tensor_operation::element_wise::Add_Mul_Activation_Mul_Clamp< Activation >::operator() ( int32_t & y,
const int32_t & x,
const int32_t & bias ) const
inlineconstexpr

◆ operator()() [2/2]

template<typename Activation>
__host__ __device__ constexpr void ck::tensor_operation::element_wise::Add_Mul_Activation_Mul_Clamp< Activation >::operator() ( int8_t & y,
const int32_t & x,
const int32_t & bias ) const
inlineconstexpr

Member Data Documentation

◆ activationOp_

template<typename Activation>
Activation ck::tensor_operation::element_wise::Add_Mul_Activation_Mul_Clamp< Activation >::activationOp_

◆ name

template<typename Activation>
const char* ck::tensor_operation::element_wise::Add_Mul_Activation_Mul_Clamp< Activation >::name = "Add_Mul_Activation_Mul_Clamp"
staticconstexpr

◆ scale_z_inv_

template<typename Activation>
float ck::tensor_operation::element_wise::Add_Mul_Activation_Mul_Clamp< Activation >::scale_z_inv_

◆ scaleAcc_

template<typename Activation>
float ck::tensor_operation::element_wise::Add_Mul_Activation_Mul_Clamp< Activation >::scaleAcc_
The documentation for this struct was generated from the following file: