List of all items

Structs

• ops::apply_rope::ApplyRope
• ops::apply_rope::RotateHalf
• ops::dyn_kv_cache::DynKeyValueCache
• ops::dyn_kv_cache::DynKeyValueCacheState
• ops::dyn_kv_cache::FrozenDynKeyValueCacheState
• ops::flash_sdpa::FlashSdpaOp
• ops::gelu_approximate::GeluApproximate
• ops::rms_norm::RmsNorm
• ops::scaled_masked_softmax::ScaledMaskedSoftmax
• ops::sdpa::Sdpa
• ops::silu::Silu
• ops::streamed_sdpa::StreamedSdpaOp

Enums

• ops::sdpa::SdpaMaskMode

Traits

• WithTractTransformers

Functions

• figure_out_causal_llm_b_s_p
• memory_arena_hints_for_causal_llm
• ops::apply_rope::apply_rope_rule
• ops::apply_rope::register
• ops::apply_rope::rotate_half_rule
• ops::dyn_kv_cache::register
• ops::dyn_kv_cache::replace_kv_cache
• ops::dyn_kv_cache::unfold_kv_cache
• ops::gelu_approximate::detect_gelu_approx
• ops::gelu_approximate::gelu_approximate
• ops::rms_norm::detect_rms_norm
• ops::scaled_masked_softmax::register
• ops::scaled_masked_softmax::scaled_masked_softmax_rule
• ops::sdpa::fuse_kv_cache_broadcast_rule
• ops::sdpa::match_broadcast_kv_cache_pattern
• ops::sdpa::register
• ops::sdpa::wire_attention_mask
• ops::silu::detect_silu
• ops::silu::silu
• register
• tract_transformers_registry