Class ShapeStructInfo

python/tvm/relax/struct_info.py:110–138

StructInfo of a shape value. Parameters ---------- values : Optional[List[PrimExpr]] The symbolic shape values if known. ndim : Optional[int] The size of the shape. Note ---- Do not specify values and ndim at the same time.

@tvm_ffi.register_object("relax.ShapeStructInfo")
class ShapeStructInfo(StructInfo):
    """StructInfo of a shape value.

    Parameters
    ----------
    values : Optional[List[PrimExpr]]
       The symbolic shape values if known.

    ndim : Optional[int]
       The size of the shape.

    Note
    ----
    Do not specify values and ndim at the same time.
    """

    values: list[PrimExpr] | None
    ndim: int
    span: Span

    def __init__(
        self, values: list[PrimExpr] | None = None, ndim: int = -1, span: Span = None
    ) -> None:
        self.__init_handle_by_constructor__(
            _ffi_api.ShapeStructInfo,
            values,
            ndim,
            span,  # type: ignore
        )


@tvm_ffi.register_object("relax.TensorStructInfo")