/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/develop/include/ck/tensor_operation/gpu/element/quantization_operation.hpp Source File#
quantization_operation.hpp
Go to the documentation of this file.
98 // Conv Perchannel quantization + Activation function which is piecewise linear function, such as
173 // Conv Perchannel quantization + Activation function which is piecewise linear function, such as
__host__ constexpr __device__ T clamp(const T &x, const T &lowerbound, const T &upperbound)
Definition: math.hpp:148
Definition: ck.hpp:267
Definition: quantization_operation.hpp:103
Activation activationOp_
Definition: quantization_operation.hpp:126
constexpr __device__ void operator()(int32_t &y, const int32_t &x, const float &requantScale) const
Definition: quantization_operation.hpp:116
Activation_Mul2_Clamp(Activation activationOp)
Definition: quantization_operation.hpp:104
__host__ constexpr __device__ void operator()(int8_t &y, const int32_t &x, const float &requantScale) const
Definition: quantization_operation.hpp:107
Definition: quantization_operation.hpp:26
Activation activationOp_
Definition: quantization_operation.hpp:65
float requantScale_
Definition: quantization_operation.hpp:64
constexpr __host__ void operator()(float &y, const float &x) const
Definition: quantization_operation.hpp:57
__host__ constexpr __device__ void operator()(int8_t &y, const int32_t &x) const
Definition: quantization_operation.hpp:39
Activation_Mul_Clamp(float requantScale, Activation activationOp)
Definition: quantization_operation.hpp:34
constexpr __device__ void operator()(int32_t &y, const int32_t &x) const
Definition: quantization_operation.hpp:47
Definition: quantization_operation.hpp:177
__host__ constexpr __device__ void operator()(int8_t &y, const int32_t &x, const int32_t &bias, const float &requantScale) const
Definition: quantization_operation.hpp:181
__host__ constexpr __device__ void operator()(int32_t &y, const int32_t &x, const int32_t &bias, const float &requantScale) const
Definition: quantization_operation.hpp:190
Add_Activation_Mul2_Clamp(Activation activationOp)
Definition: quantization_operation.hpp:178
Activation activationOp_
Definition: quantization_operation.hpp:200
Definition: quantization_operation.hpp:133
Activation activationOp_
Definition: quantization_operation.hpp:170
float requantScale_
Definition: quantization_operation.hpp:169
__host__ constexpr __device__ void operator()(int32_t &y, const int32_t &x, const int32_t &bias) const
Definition: quantization_operation.hpp:159
Add_Activation_Mul_Clamp(float requantScale, Activation activationOp)
Definition: quantization_operation.hpp:144
__host__ constexpr __device__ void operator()(int8_t &y, const int32_t &x, const int32_t &bias) const
Definition: quantization_operation.hpp:150
Definition: quantization_operation.hpp:252
Activation activationOp_
Definition: quantization_operation.hpp:281
__host__ constexpr __device__ void operator()(int32_t &y, const int32_t &x, const int32_t &bias, const float &scaleAcc) const
Definition: quantization_operation.hpp:269
__host__ constexpr __device__ void operator()(int8_t &y, const int32_t &x, const int32_t &bias, const float &scaleAcc) const
Definition: quantization_operation.hpp:259
float scale_z_inv_
Definition: quantization_operation.hpp:280
Add_Mul2_Activation_Mul_Clamp(float scale_z_inv, Activation activationOp)
Definition: quantization_operation.hpp:253
Definition: quantization_operation.hpp:208
float scaleAcc_
Definition: quantization_operation.hpp:242
__host__ constexpr __device__ void operator()(int32_t &y, const int32_t &x, const int32_t &bias) const
Definition: quantization_operation.hpp:230
Add_Mul_Activation_Mul_Clamp(float scale_z_inv, float scaleAcc, Activation activationOp)
Definition: quantization_operation.hpp:214
float scale_z_inv_
Definition: quantization_operation.hpp:241
Activation activationOp_
Definition: quantization_operation.hpp:243
__host__ constexpr __device__ void operator()(int8_t &y, const int32_t &x, const int32_t &bias) const
Definition: quantization_operation.hpp:220
Definition: quantization_operation.hpp:73
float scaleAcc_
Definition: quantization_operation.hpp:94
Mul_Activation_Mul_Clamp(float scale_z_inv, float scaleAcc, Activation activationOp)
Definition: quantization_operation.hpp:79
float scale_z_inv_
Definition: quantization_operation.hpp:93
Activation activationOp_
Definition: quantization_operation.hpp:95
__host__ constexpr __device__ void operator()(int8_t &y, const int32_t &x) const
Definition: quantization_operation.hpp:84