List of all items

Structs

• Context
• Tensor

Functions

• cpu::add
• cpu::attention
• cpu::div
• cpu::gelu
• cpu::layernorm
• cpu::matmul_bt
• cpu::mha_causal
• cpu::relu
• cpu::rmsnorm
• cpu::rope
• cpu::sigmoid
• cpu::silu
• cpu::softmax
• cpu::sqrt
• cpu::sub
• demo::generate
• demo::logits
• demo::logits_cpu
• matmul_cpu
• max_abs_diff

Type Aliases

• Result

Constants

• demo::BASE
• demo::D
• demo::DH
• demo::EPS
• demo::H
• demo::HFF
• demo::N_LAYERS
• demo::VOCAB