layernorm2d_fwd_traits.hpp File Reference

layernorm2d_fwd_traits.hpp File Reference#

Composable Kernel: layernorm2d_fwd_traits.hpp File Reference
Classes | Namespaces | Enumerations
layernorm2d_fwd_traits.hpp File Reference
#include "ck_tile/core/utility/type_traits.hpp"

Go to the source code of this file.

Classes

struct  ck_tile::Layernorm2dXBiasEnumName< Layernorm2dXBiasEnum::NO_BIAS >
struct  ck_tile::Layernorm2dXBiasEnumName< Layernorm2dXBiasEnum::ADD_BIAS >
struct  ck_tile::Layernorm2dFusedAddEnumName< Layernorm2dFusedAddEnum::NO_ADD >
struct  ck_tile::Layernorm2dFusedAddEnumName< Layernorm2dFusedAddEnum::PRE_ADD_STORE >
struct  ck_tile::Layernorm2dFusedAddEnumName< Layernorm2dFusedAddEnum::PRE_ADD >
struct  ck_tile::Layernorm2dFusedQuantEnumName< Layernorm2dFusedQuantEnum::NO_SWEEP >
struct  ck_tile::Layernorm2dFusedQuantEnumName< Layernorm2dFusedQuantEnum::DYNAMIC_QUANT >
struct  ck_tile::Layernorm2dFusedQuantEnumName< Layernorm2dFusedQuantEnum::SMOOTH_DYNAMIC_QUANT >
struct  ck_tile::Layernorm2dFwdTraits< kPadN_, kSaveMeanInvStd_, kFastFDiv_, kWelford_, kTwoPass_, kXbias_, kFusedAdd_, kFusedQuant_ >

Namespaces

namespace  ck_tile

Enumerations

enum class  ck_tile::Layernorm2dXBiasEnum {
  ck_tile::NO_BIAS = 0 ,
  ck_tile::ADD_BIAS = 1
}
enum class  ck_tile::Layernorm2dFusedAddEnum {
  ck_tile::NO_ADD = 0 ,
  ck_tile::PRE_ADD_STORE = 1 ,
  ck_tile::PRE_ADD = 2
}
enum class  ck_tile::Layernorm2dFusedQuantEnum {
  ck_tile::NO_SWEEP = 0 ,
  ck_tile::SMOOTH_DYNAMIC_QUANT = 1 ,
  ck_tile::DYNAMIC_QUANT = 2
}