unary_element_wise_operation.hpp File Reference

• include
• ck
• tensor_operation
• gpu
• element

Classes | Namespaces | Functions

unary_element_wise_operation.hpp File Reference

#include "ck/utility/data_type.hpp"
#include "ck/utility/math.hpp"
#include "ck/utility/math_v2.hpp"
#include "ck/utility/type_convert.hpp"
#include "ck/utility/amd_inline_asm.hpp"
#include <cassert>

Go to the source code of this file.

Classes
struct	ck::tensor_operation::element_wise::PassThroughPack8
struct	ck::tensor_operation::element_wise::DequantPack8
struct	ck::tensor_operation::element_wise::PassThroughPack2
struct	ck::tensor_operation::element_wise::PassThrough
struct	ck::tensor_operation::element_wise::UnaryConvert
struct	ck::tensor_operation::element_wise::ConvertBF16RTN
struct	ck::tensor_operation::element_wise::ConvertF8SR
struct	ck::tensor_operation::element_wise::ConvertF8RNE
struct	ck::tensor_operation::element_wise::Scale
struct	ck::tensor_operation::element_wise::ScaleAndResetNaNToMinusInfinity
struct	ck::tensor_operation::element_wise::UnaryDivide
struct	ck::tensor_operation::element_wise::UnarySquare
struct	ck::tensor_operation::element_wise::UnaryAbs
struct	ck::tensor_operation::element_wise::UnarySqrt
struct	ck::tensor_operation::element_wise::Clamp
struct	ck::tensor_operation::element_wise::Relu
struct	ck::tensor_operation::element_wise::FastGelu
struct	ck::tensor_operation::element_wise::Gelu
struct	ck::tensor_operation::element_wise::Sigmoid
struct	ck::tensor_operation::element_wise::Silu
struct	ck::tensor_operation::element_wise::TanH
struct	ck::tensor_operation::element_wise::ACos
struct	ck::tensor_operation::element_wise::Neg
struct	ck::tensor_operation::element_wise::ATan
struct	ck::tensor_operation::element_wise::Sin
struct	ck::tensor_operation::element_wise::ASinH
struct	ck::tensor_operation::element_wise::Cos
struct	ck::tensor_operation::element_wise::ACosH
struct	ck::tensor_operation::element_wise::Tan
struct	ck::tensor_operation::element_wise::ATanH
struct	ck::tensor_operation::element_wise::SinH
struct	ck::tensor_operation::element_wise::Ceil
struct	ck::tensor_operation::element_wise::Exp
struct	ck::tensor_operation::element_wise::CosH
struct	ck::tensor_operation::element_wise::Floor
struct	ck::tensor_operation::element_wise::Log
struct	ck::tensor_operation::element_wise::ASin
struct	ck::tensor_operation::element_wise::Rcp
struct	ck::tensor_operation::element_wise::Swish
struct	ck::tensor_operation::element_wise::SoftRelu
struct	ck::tensor_operation::element_wise::Power
struct	ck::tensor_operation::element_wise::ClippedRelu
struct	ck::tensor_operation::element_wise::LeakyRelu
struct	ck::tensor_operation::element_wise::Elu
struct	ck::tensor_operation::element_wise::Logistic
struct	ck::tensor_operation::element_wise::ConvInvscale
struct	ck::tensor_operation::element_wise::ConvScale
struct	ck::tensor_operation::element_wise::ConvScaleRelu
struct	ck::tensor_operation::element_wise::FastNumericArrayConverter< InputDataType, OutputDataType, RegPackNumber >
struct	ck::tensor_operation::element_wise::FastNumericArrayConverter< uint8_t, half_t, 4 >
struct	ck::tensor_operation::element_wise::FastNumericArrayConverter< uint8_t, half_t, N >
struct	ck::tensor_operation::element_wise::DynamicUnaryOp

Namespaces
namespace	ck
namespace	ck::tensor_operation
namespace	ck::tensor_operation::element_wise

Functions
__device__ half4_t	ck::i4_to_half4 (int q)
__device__ half4_t	ck::i4_to_half4_scale (int q, const ck::half2_t &scale)
__device__ f8x4_t	ck::i4_to_f8x4 (int q)
__device__ f8x8_t	ck::i4_to_fp8x8 (int q)
__device__ bhalf4_t	ck::i4_to_bhalf4 (int q)