List of all items

Structs

• kernels::attention::CompiledDenseCausalAttentionPlan
• kernels::projection::LowrankForwardRouteProfileSnapshot
• kernels::recurrent::CompiledRecurrentAttentionPlan
• kernels::recurrent::RecurrentAttentionOutput
• kernels::sequence::linear::CompiledDenseCausalAttentionPlan
• kernels::sequence::linear::CompiledRecurrentAttentionPlan
• kernels::sequence::linear::RecurrentAttentionOutput
• kernels::sequence::mamba3::forward::Mamba3TensorizedOutput
• kernels::sequence::mamba3::forward::Mamba3TensorizedState

Enums

• kernels::projection::LowrankGradInputExecutor

Functions

• kernels::attention::supports_dense_causal_attention_backend
• kernels::attention::try_fused_dense_causal_attention_wgpu
• kernels::attention::try_fused_dense_causal_attention_wgpu_with_plan
• kernels::projection::relu_lowrank_forward_profile_reset
• kernels::projection::relu_lowrank_forward_profile_snapshot
• kernels::projection::relu_lowrank_forward_route_profile_reset
• kernels::projection::relu_lowrank_forward_route_profile_snapshot
• kernels::projection::relu_lowrank_grad_input_profile_reset
• kernels::projection::relu_lowrank_grad_input_profile_snapshot
• kernels::projection::relu_lowrank_grad_weight_profile_reset
• kernels::projection::relu_lowrank_grad_weight_profile_snapshot
• kernels::projection::supports_relu_lowrank_projection_backend
• kernels::projection::try_fused_relu_lowrank_projection_wgpu
• kernels::projection::try_fused_relu_lowrank_projection_wgpu_with_executor
• kernels::recurrent::recurrent_profile_reset
• kernels::recurrent::recurrent_profile_snapshot
• kernels::recurrent::supports_recurrent_backend
• kernels::recurrent::try_fused_recurrent_attention_wgpu
• kernels::recurrent::try_fused_recurrent_attention_wgpu_with_plan
• kernels::sequence::linear::recurrent_profile_reset
• kernels::sequence::linear::recurrent_profile_snapshot
• kernels::sequence::linear::supports_dense_causal_attention_backend
• kernels::sequence::linear::supports_recurrent_backend
• kernels::sequence::linear::try_fused_dense_causal_attention_wgpu
• kernels::sequence::linear::try_fused_dense_causal_attention_wgpu_with_plan
• kernels::sequence::linear::try_fused_recurrent_attention_wgpu
• kernels::sequence::linear::try_fused_recurrent_attention_wgpu_with_plan
• kernels::sequence::mamba3::forward::tensorized_mamba3_forward
• kernels::sequence::mamba3::forward::tensorized_mamba3_forward_custom_backward
• kernels::sequence::mamba3::forward::tensorized_mamba3_forward_direct_graph
• kernels::sequence::mamba3::forward::use_tensorized_mamba3_forward_experimental

Type Aliases

• kernels::projection::LowrankProjectionProfileSnapshot
• kernels::recurrent::RecurrentProfileSnapshot
• kernels::sequence::linear::RecurrentProfileSnapshot

Constants

• kernels::sequence::mamba3::BACKWARD_ACCELERATION_AVAILABLE
• kernels::sequence::mamba3::FORWARD_ACCELERATION_AVAILABLE
• kernels::sequence::mamba3::STATUS
• kernels::sequence::mamba3::UPSTREAM_REPO
• kernels::sequence::mamba3::UPSTREAM_TARGET_KIND
• kernels::sequence::mamba3::backward::AVAILABLE