layer_norm_fused_op_api.hpp File Reference

C++ API. More...

#include "subgroup/subgroup.hpp"

Include dependency graph for layer_norm_fused_op_api.hpp:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes
struct	gpu::xetla::group::ln_fwd_fused_op_t< fused_op_kind_, dtype_in_, dtype_out_, dtype_acc_, layer_norm_attr_, arch_ >
struct	gpu::xetla::group::ln_bwd_fused_op_t< fused_op_kind_, dtype_in_, dtype_out_, dtype_acc_, layer_norm_attr_, arch_ >

Namespaces
namespace	gpu
namespace	gpu::xetla
namespace	gpu::xetla::group

Enumerations
enum class	gpu::xetla::ln_fwd_fused_kind : uint8_t {   gpu::xetla::none = 0 , gpu::xetla::bias_dropout_resAdd_ln = 1 , gpu::xetla::ln_dropout = 2 , gpu::xetla::bias_rng_dropout_resAdd_ln = 3 ,   gpu::xetla::ln_rng_dropout = 4 }
enum class	gpu::xetla::ln_bwd_fused_kind : uint8_t { gpu::xetla::none = 0 , gpu::xetla::bias_dropout_resAdd_ln = 1 , gpu::xetla::ln_dropout_gradAdd = 2 , gpu::xetla::ln_dropout = 3 }

Detailed Description

C++ API.