Struct TensorSpec 

pub struct TensorSpec {
    pub name: &'static str,
    pub dtype: KdaDtype,
    /* private fields */
}

The 15 self_attn tensors a KDA block has — and the complete list of what it may have.

Shapes are expressed against Glm5NextKdaConfig so a geometry change fails loudly here rather than at launch. H = heads, D = head_dim, Q = H*D, X = hidden, K = conv kernel.

Fields

name: &'static str

dtype: KdaDtype

Implementations

impl TensorSpec

pub fn expected_shape(&self, c: &Glm5NextKdaConfig) -> Vec<usize>

Trait Implementations

impl Clone for TensorSpec

fn clone(&self) -> TensorSpec

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for TensorSpec

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

Formats the value using the given formatter.

impl Copy for TensorSpec