Skip to main content

/

Python module

shape

Library for graph shape types.

`Shape`

class max.graph.shape.Shape(dims=())

Parameters:: dims (ShapeLike)

`from_mlir()`

classmethod from_mlir(attr)

Parameters:: attr (TypedAttr)
Return type:: Shape

`is_static()`

static is_static(shape)

Parameters:: shape (Shape)
Return type:: TypeGuard[list[StaticDim]]

`parameters`

property parameters: Iterable[SymbolicDim]

Lists the symbolic dimension names on which this shape depends.

`rank`

property rank

`static_dims`

property static_dims: list[int]

Returns all static dims in the shape as a list of integers.

`to_mlir()`

to_mlir()

Return type:: ShapeAttr

Shape

View source

View source

Was this page helpful?

Thank you! We'll create more content like this.

Thank you for helping us improve!