Layernorm2dFwdHostArgs Struct Reference

Layernorm2dFwdHostArgs Struct Reference#

Composable Kernel: ck_tile::Layernorm2dFwdHostArgs Struct Reference
Public Attributes | List of all members
ck_tile::Layernorm2dFwdHostArgs Struct Reference

#include <layernorm2d_fwd_kernel.hpp>

Public Attributes

const void * p_x
const void * p_x_residual
const void * p_sm_scale
const void * p_x_bias
const void * p_gamma
const void * p_beta
void * p_y
void * p_y_residual
void * p_y_scale
void * p_mean
void * p_invStd
float epsilon
index_t m
index_t n
index_t x_stride
index_t xr_stride
index_t y_stride
index_t yr_stride

Member Data Documentation

◆ epsilon

float ck_tile::Layernorm2dFwdHostArgs::epsilon

◆ m

index_t ck_tile::Layernorm2dFwdHostArgs::m

◆ n

index_t ck_tile::Layernorm2dFwdHostArgs::n

◆ p_beta

const void* ck_tile::Layernorm2dFwdHostArgs::p_beta

◆ p_gamma

const void* ck_tile::Layernorm2dFwdHostArgs::p_gamma

◆ p_invStd

void* ck_tile::Layernorm2dFwdHostArgs::p_invStd

◆ p_mean

void* ck_tile::Layernorm2dFwdHostArgs::p_mean

◆ p_sm_scale

const void* ck_tile::Layernorm2dFwdHostArgs::p_sm_scale

◆ p_x

const void* ck_tile::Layernorm2dFwdHostArgs::p_x

◆ p_x_bias

const void* ck_tile::Layernorm2dFwdHostArgs::p_x_bias

◆ p_x_residual

const void* ck_tile::Layernorm2dFwdHostArgs::p_x_residual

◆ p_y

void* ck_tile::Layernorm2dFwdHostArgs::p_y

◆ p_y_residual

void* ck_tile::Layernorm2dFwdHostArgs::p_y_residual

◆ p_y_scale

void* ck_tile::Layernorm2dFwdHostArgs::p_y_scale

◆ x_stride

index_t ck_tile::Layernorm2dFwdHostArgs::x_stride

◆ xr_stride

index_t ck_tile::Layernorm2dFwdHostArgs::xr_stride

◆ y_stride

index_t ck_tile::Layernorm2dFwdHostArgs::y_stride

◆ yr_stride

index_t ck_tile::Layernorm2dFwdHostArgs::yr_stride
The documentation for this struct was generated from the following file: