element_wise Namespace Reference

element_wise Namespace Reference#

Composable Kernel: ck::tensor_operation::element_wise Namespace Reference
Classes | Typedefs
ck::tensor_operation::element_wise Namespace Reference

Classes

struct  Add
struct  Max
struct  Min
struct  Multiply
struct  ScaleAdd
struct  Subtract
struct  Bilinear
struct  AddClamp
struct  AddRelu
struct  AddHardswish
struct  AddFastGelu
struct  MultiplyFastGelu
struct  AddSilu
struct  ConvScaleAdd
struct  UnaryCombinedOp
struct  BinaryWithUnaryCombinedOp
struct  TrinaryWithUnaryCombinedOp
struct  AddReluAdd
struct  AddHardswishAdd
struct  AddAdd
struct  AddMultiply
struct  MultiplyAdd
struct  MultiplyMultiply
struct  MultiplyAddFastGelu
struct  AddAddFastGelu
struct  ScaleAddScaleAddRelu
struct  Normalize
struct  NormalizeInInfer
struct  BiasNormalizeInInferClamp
struct  UnaryTypeConvert
struct  UnaryTypeConvert< float, ck::bhalf_t >
struct  UnaryTypeConvert< ck::bhalf_t, float >
struct  Activation_Mul_Clamp
struct  Mul_Activation_Mul_Clamp
struct  Activation_Mul2_Clamp
struct  Add_Activation_Mul_Clamp
struct  Add_Activation_Mul2_Clamp
struct  Add_Mul_Activation_Mul_Clamp
struct  Add_Mul2_Activation_Mul_Clamp
struct  PassThroughPack8
struct  DequantPack8
struct  PassThroughPack2
struct  PassThrough
struct  UnaryConvert
struct  ConvertBF16RTN
struct  ConvertF8SR
struct  ConvertF8RNE
struct  Scale
struct  ScaleAndResetNaNToMinusInfinity
struct  UnaryDivide
struct  UnarySquare
struct  UnaryAbs
struct  UnarySqrt
struct  Clamp
struct  Relu
struct  FastGelu
struct  Gelu
struct  Sigmoid
struct  Silu
struct  TanH
struct  ACos
struct  Neg
struct  ATan
struct  Sin
struct  ASinH
struct  Cos
struct  ACosH
struct  Tan
struct  ATanH
struct  SinH
struct  Ceil
struct  Exp
struct  CosH
struct  Floor
struct  Log
struct  ASin
struct  Rcp
struct  Swish
struct  SoftRelu
struct  Power
struct  ClippedRelu
struct  LeakyRelu
struct  Elu
struct  Logistic
struct  ConvInvscale
struct  ConvScale
struct  ConvScaleRelu
struct  FastNumericArrayConverter
struct  FastNumericArrayConverter< uint8_t, half_t, 4 >
struct  FastNumericArrayConverter< uint8_t, half_t, N >
struct  DynamicUnaryOp

Typedefs

using ScaleScalePass = UnaryCombinedOp<Scale, Scale, PassThrough>
using ScaleScaleRelu = UnaryCombinedOp<Scale, Scale, Relu>

Typedef Documentation

◆ ScaleScalePass

using ck::tensor_operation::element_wise::ScaleScalePass = UnaryCombinedOp<Scale, Scale, PassThrough>

◆ ScaleScaleRelu

using ck::tensor_operation::element_wise::ScaleScaleRelu = UnaryCombinedOp<Scale, Scale, Relu>