gpu::xetla::kernel::layer_norm_fwd_t< dtype_x_, dtype_y_, dtype_weight_, dtype_acc_, layer_norm_attr_, store_for_bwd_, arch_, ln_fwd_fused_op_ > Struct Template Reference

#include <api.hpp>

Detailed Description

template<typename dtype_x_, typename dtype_y_, typename dtype_weight_, typename dtype_acc_, typename layer_norm_attr_, bool store_for_bwd_ = true, gpu_arch arch_ = gpu_arch::Xe, typename ln_fwd_fused_op_ = group::ln_fwd_fused_op_t<ln_fwd_fused_kind::none, dtype_x_, dtype_y_, dtype_acc_, layer_norm_attr_, arch_>>
struct gpu::xetla::kernel::layer_norm_fwd_t< dtype_x_, dtype_y_, dtype_weight_, dtype_acc_, layer_norm_attr_, store_for_bwd_, arch_, ln_fwd_fused_op_ >

Template Parameters

dtype_x_
dtype_y_
dtype_weight_
dtype_acc_
layer_norm_attr_
store_for_bwd_
arch_
ln_fwd_fused_op_