Python class

StaticDim

`StaticDim`

class max.graph.StaticDim(value)

source

Bases: Dim

A static tensor dimension with a fixed size.

Because a static dimension’s size is fixed, related computation can be optimized at compile time. This is key to good model performance.

The following example creates static dimensions implicitly by passing integer values to TensorType:

from max.graph import TensorType
from max.dtype import DType
tensor = TensorType(DType.int64, (4, 5))

Converts valid input values to Dim.

dim: int

source

The size of the static dimension.

static from_mlir(attr)

source

Constructs a StaticDim from a builtin.IntegerAttr.

Parameters:: attr (TypedAttr) – The builtin.IntegerAttr to parse into a StaticDim.
Returns:: The StaticDim represented by the builtin.IntegerAttr.
Return type:: StaticDim

property parameters: Iterable[SymbolicDim]

source

Lists the symbolic dimension names on which this dim depends.

to_mlir()

source

Creates an mlir.Attribute representing this dimension.

This is used internally when constructing tensor MLIR types.