oa::ml::matrix::SsmConfig

Explicit tensor geometry and optional paths for a selective-state scan. A zero num_groups means one Q/K group per head, matching OA C++.

StRust public API

Declaration

pub struct SsmConfig { pub batch_size: usize, pub sequence_length: usize, pub num_heads: usize, pub num_groups: usize, pub head_dim: usize, pub state_size: usize, pub num_rope_angles: usize, pub mimo_rank: usize, pub has_z: bool, pub has_d: bool, pub has_output_norm: bool, }

Public Rust declaration.

Public Methods

fn clone(&self) -> SsmConfig

fn fmt(&self, f: &mut Formatter<'_>) -> Result

fn eq(&self, other: &SsmConfig) -> bool

Public Data Members

batch_sizeusizefield

sequence_lengthusizefield

num_headsusizefield

num_groupsusizefield

head_dimusizefield

state_sizeusizefield

num_rope_anglesusizefield

mimo_rankusizefield

has_zboolfield

has_dboolfield

has_output_normboolfield

Public Method Documentation

fn clone(
	&self
) -> SsmConfig

No public source comment is attached to this declaration.

fn fmt(
	&self,
	f: &mut Formatter<'_>
) -> Result

No public source comment is attached to this declaration.

fn eq(
	&self,
	other: &SsmConfig
) -> bool

No public source comment is attached to this declaration.