Struct DiscretizationCache 

pub struct DiscretizationCache { /* private fields */ }

Cache for discretized SSM matrices to avoid recomputation

Implementations

impl DiscretizationCache

pub fn new(num_layers: usize, hidden_dim: usize, state_dim: usize) -> Self

Create a new discretization cache

pub fn update( &mut self, layer_idx: usize, delta: f32, a_bar: Array2<f32>, b_bar: Array2<f32>, )

Update the cache with new discretized matrices

pub fn get( &self, layer_idx: usize, delta: f32, ) -> Option<(&Array2<f32>, &Array2<f32>)>

Get cached discretized matrices if valid

pub fn invalidate(&mut self)

Invalidate the cache

pub fn is_valid(&self, delta: f32) -> bool

Check if cache is valid for given delta

Trait Implementations

impl Debug for DiscretizationCache

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

Formats the value using the given formatter.