Enum DimExpr 

pub enum DimExpr {
    Const(usize),
    Var(String),
    Add(Box<DimExpr>, Box<DimExpr>),
    Sub(Box<DimExpr>, Box<DimExpr>),
    Mul(Box<DimExpr>, Box<DimExpr>),
    Div(Box<DimExpr>, Box<DimExpr>),
    Max(Box<DimExpr>, Box<DimExpr>),
    Min(Box<DimExpr>, Box<DimExpr>),
    CeilDiv(Box<DimExpr>, Box<DimExpr>),
}

A dimension expression that can be evaluated.

Variants

Const(usize)

A concrete dimension value

Var(String)

A dimension variable

Add(Box<DimExpr>, Box<DimExpr>)

Addition of dimensions

Sub(Box<DimExpr>, Box<DimExpr>)

Subtraction of dimensions

Mul(Box<DimExpr>, Box<DimExpr>)

Multiplication of dimensions

Div(Box<DimExpr>, Box<DimExpr>)

Division of dimensions (integer division)

Max(Box<DimExpr>, Box<DimExpr>)

Maximum of two dimensions

Min(Box<DimExpr>, Box<DimExpr>)

Minimum of two dimensions

CeilDiv(Box<DimExpr>, Box<DimExpr>)

Ceiling division (useful for padding/strides)

Implementations

impl DimExpr

pub fn constant(value: usize) -> Self

Create a constant dimension.

pub fn var(name: impl Into<String>) -> Self

Create a dimension variable.

pub fn add(self, other: DimExpr) -> Self

Add two dimension expressions.

pub fn sub(self, other: DimExpr) -> Self

Subtract a dimension expression from this one.

pub fn mul(self, other: DimExpr) -> Self

Multiply two dimension expressions.

pub fn div(self, other: DimExpr) -> Self

Divide this dimension expression by another.

pub fn max(self, other: DimExpr) -> Self

Take the maximum of two dimension expressions.

pub fn min(self, other: DimExpr) -> Self

Take the minimum of two dimension expressions.

pub fn ceil_div(self, other: DimExpr) -> Self

Ceiling division.

pub fn eval(&self, ctx: &DependentTypeContext) -> Option<usize>

Evaluate the dimension expression in a context.

pub fn free_variables(&self) -> Vec<String>

Get all free variables in this expression.

pub fn substitute(&self, var: &str, expr: &DimExpr) -> DimExpr

Substitute a variable with an expression.

pub fn simplify(&self) -> DimExpr

Simplify the expression.

pub fn is_equal(&self, other: &DimExpr, ctx: &DependentTypeContext) -> bool

Check if two dimension expressions are equal.

Trait Implementations

impl Clone for DimExpr

fn clone(&self) -> DimExpr

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for DimExpr

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

Formats the value using the given formatter.

impl Display for DimExpr

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

Formats the value using the given formatter.

impl PartialEq for DimExpr

fn eq(&self, other: &DimExpr) -> 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 StructuralPartialEq for DimExpr