gpu::xetla::group::ln_bwd_fused_op_arguments_t< dtype_in, dtype_out, dtype_acc > Struct Template Reference

#include <layer_norm_fused_op_bwd_xe.hpp>

Public Attributes
dtype_acc *	dbias_acc_ptr
dtype_out *	dx_resAdd_ptr
dtype_in *	gradAdd_ptr
uint8_t *	mask_ptr
uint32_t	matrix_m
uint32_t	matrix_n
uint32_t	mat_ld
uint32_t	mask_ld
float	dropout_prob
float	dropout_scale_inv

Detailed Description

template<typename dtype_in, typename dtype_out, typename dtype_acc>
struct gpu::xetla::group::ln_bwd_fused_op_arguments_t< dtype_in, dtype_out, dtype_acc >

Template Parameters

dtype_in
dtype_out
dtype_acc

Member Data Documentation

dbias_acc_ptr

template<typename dtype_in , typename dtype_out , typename dtype_acc >

dtype_acc* gpu::xetla::group::ln_bwd_fused_op_arguments_t< dtype_in, dtype_out, dtype_acc >::dbias_acc_ptr

dropout_prob

template<typename dtype_in , typename dtype_out , typename dtype_acc >

float gpu::xetla::group::ln_bwd_fused_op_arguments_t< dtype_in, dtype_out, dtype_acc >::dropout_prob

dropout_scale_inv

template<typename dtype_in , typename dtype_out , typename dtype_acc >

float gpu::xetla::group::ln_bwd_fused_op_arguments_t< dtype_in, dtype_out, dtype_acc >::dropout_scale_inv

dx_resAdd_ptr

template<typename dtype_in , typename dtype_out , typename dtype_acc >

dtype_out* gpu::xetla::group::ln_bwd_fused_op_arguments_t< dtype_in, dtype_out, dtype_acc >::dx_resAdd_ptr

gradAdd_ptr

template<typename dtype_in , typename dtype_out , typename dtype_acc >

dtype_in* gpu::xetla::group::ln_bwd_fused_op_arguments_t< dtype_in, dtype_out, dtype_acc >::gradAdd_ptr

mask_ld

template<typename dtype_in , typename dtype_out , typename dtype_acc >

uint32_t gpu::xetla::group::ln_bwd_fused_op_arguments_t< dtype_in, dtype_out, dtype_acc >::mask_ld

mask_ptr

template<typename dtype_in , typename dtype_out , typename dtype_acc >

uint8_t* gpu::xetla::group::ln_bwd_fused_op_arguments_t< dtype_in, dtype_out, dtype_acc >::mask_ptr

mat_ld

template<typename dtype_in , typename dtype_out , typename dtype_acc >

uint32_t gpu::xetla::group::ln_bwd_fused_op_arguments_t< dtype_in, dtype_out, dtype_acc >::mat_ld

matrix_m

template<typename dtype_in , typename dtype_out , typename dtype_acc >

uint32_t gpu::xetla::group::ln_bwd_fused_op_arguments_t< dtype_in, dtype_out, dtype_acc >::matrix_m

matrix_n

template<typename dtype_in , typename dtype_out , typename dtype_acc >

uint32_t gpu::xetla::group::ln_bwd_fused_op_arguments_t< dtype_in, dtype_out, dtype_acc >::matrix_n