Class Var

python/tvm/tirx/expr.py:355–374

Symbolic variable. Parameters ---------- name : str The name dtype : Union[str, ir.Type] The data type span : Optional[Span] The location of this expression in the source code.

@tvm_ffi.register_object("tirx.Var")
class Var(PrimExprWithOp):
    """Symbolic variable.

    Parameters
    ----------
    name : str
        The name

    dtype : Union[str, ir.Type]
        The data type

    span : Optional[Span]
        The location of this expression in the source code.
    """

    name_hint: str
    type_annotation: ir.Type

    def __init__(self, name: str, dtype: str | ir.Type, span: Span | None = None) -> None:
        self.__init_handle_by_constructor__(_ffi_api.Var, name, dtype, span)  # type: ignore


@tvm_ffi.register_object("tirx.SizeVar")