Activation_Mul_Clamp< Activation > Struct Template Reference

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

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

#include <quantization_operation.hpp>

Public Member Functions

 Activation_Mul_Clamp (float requantScale, Activation activationOp)
__host__ __device__ constexpr void operator() (int8_t &y, const int32_t &x) const
__device__ constexpr void operator() (int32_t &y, const int32_t &x) const
__host__ constexpr void operator() (float &y, const float &x) const

Public Attributes

float requantScale_
Activation activationOp_

Static Public Attributes

static constexpr const char * name = "Activation_Mul_Clamp"

Constructor & Destructor Documentation

◆ Activation_Mul_Clamp()

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

Member Function Documentation

◆ operator()() [1/3]

template<typename Activation>
__host__ constexpr void ck::tensor_operation::element_wise::Activation_Mul_Clamp< Activation >::operator() ( float & y,
const float & x ) const
inlineconstexpr

◆ operator()() [2/3]

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

◆ operator()() [3/3]

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

Member Data Documentation

◆ activationOp_

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

◆ name

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

◆ requantScale_

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