Struct GdnScanStage 

pub struct GdnScanStage {
    pub state_size: usize,
    pub out_shape: Shape,
    pub carry_state: bool,
    pub state_key: Option<String>,
}

Q/K/V/G/Beta tensors must already be shaped [batch, seq, heads, state].

Fields

state_size: usize

out_shape: Shape

carry_state: bool

state_key: Option<String>

Implementations

impl GdnScanStage

pub fn prefill(state_size: usize, out_shape: Shape) -> Self

pub fn with_carry(self, state_key: impl Into<String>) -> Self

Trait Implementations

impl Clone for GdnScanStage

fn clone(&self) -> GdnScanStage

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for GdnScanStage

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

Formats the value using the given formatter.