Class MatchCast

python/tvm/relax/expr.py:900–932

Runtime-match the value to the struct info. This operation does runtime check, populates the un-defined symbolic shape vars and vars in struct_info in the first occurrence, and insert equality assertions in other cases. Parameters ---------- var: Var The return vari

@tvm_ffi.register_object("relax.expr.MatchCast")
class MatchCast(Binding):
    """Runtime-match the value to the struct info.

    This operation does runtime check, populates the un-defined symbolic shape vars
    and vars in struct_info in the first occurrence, and insert equality assertions in
    other cases.

    Parameters
    ----------
    var: Var
        The return variable that the match cast bind to.

    value: Expr
        The input value expression.

    struct_info: tvm.relax.StructInfo
        The struct info to match cast to.
    """

    struct_info: StructInfo
    value: Expr
    span: Span | None

    def __init__(
        self, var: Var, value: Expr, struct_info: StructInfo, span: Span | None = None
    ) -> None:
        self.__init_handle_by_constructor__(
            _ffi_api.MatchCast,
            var,
            value,
            struct_info,
            span,  # type: ignore
        )


@tvm_ffi.register_object("relax.expr.VarBinding")