List of all items

Structs

• Transformer
• attention::AttentionHead
• attention::Score
• attention::multi::Config
• attention::multi::MultiHeadAttention
• codec::Codec
• codec::CodecBuilder
• codec::Context
• codec::Generator
• codec::decoder::Decoder
• codec::decoder::config::DecoderConfig
• codec::decoder::layer::DecoderLayer
• codec::encoder::Encoder
• codec::encoder::config::EncoderConfig
• codec::encoder::layer::EncoderLayer
• model::sublayer::Sublayer
• params::QkvBase
• prelude::AttentionHead
• prelude::MultiHeadAttention
• prelude::MultiHeadAttentionConfig
• prelude::Score

Enums

• params::item::Entry
• params::item::QKV

Traits

• attention::Attention
• ops::DimMerge
• ops::DimSplit
• ops::Merge
• ops::SplitHead

Functions

• attention::scaled_dot_product_attention
• outputs_from_ratio
• prelude::scaled_dot_product_attention

Type Aliases

• params::ArcQkv
• params::Qkv
• params::ViewQkv

Constants

• DK
• D_MODEL
• D_NETWORK
• HEADS
• N
• consts::DK
• consts::D_MODEL
• consts::D_NETWORK
• consts::HEADS
• consts::N