Add_Activation_Mul_Clamp< Activation > Struct Template Reference

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

Composable Kernel: ck::tensor_operation::element_wise::Add_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_Activation_Mul_Clamp< Activation > Struct Template Reference

#include <quantization_operation.hpp>

Public Member Functions

 Add_Activation_Mul_Clamp (float requantScale, 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 requantScale_
Activation activationOp_

Static Public Attributes

static constexpr const char * name = "Add_Activation_Mul_Clamp"

Constructor & Destructor Documentation

◆ Add_Activation_Mul_Clamp()

template<typename Activation>
ck::tensor_operation::element_wise::Add_Activation_Mul_Clamp< Activation >::Add_Activation_Mul_Clamp ( float requantScale,
Activation activationOp )
inline

Member Function Documentation

◆ operator()() [1/2]

template<typename Activation>
__host__ __device__ constexpr void ck::tensor_operation::element_wise::Add_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_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_Activation_Mul_Clamp< Activation >::activationOp_

◆ name

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

◆ requantScale_

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