Enum TyNat 

pub enum TyNat {
    Lit(u64),
    Var(TyVar),
    Add(Box<TyNat>, Box<TyNat>),
    Mul(Box<TyNat>, Box<TyNat>),
}

A type-level natural number.

Represents dimensions in shape-indexed tensor types. Supports both concrete values (for static shapes) and variables (for polymorphic operations like matmul).

Variants

Lit(u64)

A concrete natural number literal.

Example: 1024 in Tensor '[1024, 768] Float

Var(TyVar)

A type-level natural variable.

Example: n in Tensor '[n] Float

Add(Box<TyNat>, Box<TyNat>)

Addition of two naturals.

Example: m + k for computing output dimensions

Mul(Box<TyNat>, Box<TyNat>)

Multiplication of two naturals.

Example: m * k for computing buffer sizes

Implementations

impl TyNat

pub fn lit(n: u64) -> Self

Creates a literal type-level natural.

pub fn zero() -> Self

Creates a zero type-level natural.

pub fn one() -> Self

Creates a one type-level natural.

pub fn add(a: TyNat, b: TyNat) -> Self

Creates an addition of two naturals.

pub fn mul(a: TyNat, b: TyNat) -> Self

Creates a multiplication of two naturals.

pub fn is_lit(&self) -> bool

Returns true if this is a concrete literal.

pub fn as_lit(&self) -> Option<u64>

Returns the literal value if this is a concrete literal.

pub fn is_ground(&self) -> bool

Returns true if this natural contains no variables.

pub fn eval(&self) -> Option<u64>

Evaluates this natural if it is ground (contains no variables).

Returns None if the natural contains variables.

pub fn free_vars(&self) -> Vec<TyVar>

Collects all type variables occurring in this natural.

pub fn subst(&self, var: &TyVar, replacement: &TyNat) -> Self

Substitutes a type variable with a natural in this expression.

Trait Implementations

impl Clone for TyNat

fn clone(&self) -> TyNat

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for TyNat

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

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for TyNat

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Display for TyNat

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

Formats the value using the given formatter.

impl Eq for TyNat

impl Hash for TyNat

fn hash<__H: Hasher>(&self, state: &mut __H)

Feeds this value into the given Hasher.

fn hash_slice<H>(data: &[Self], state: &mut H)

where H: Hasher, Self: Sized,

Feeds a slice of this type into the given Hasher.

impl PartialEq for TyNat

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

Tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &Rhs) -> bool

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl Serialize for TyNat

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.

impl StructuralPartialEq for TyNat