Function assert_op

python/tvm/relax/op/base.py:581–617

Create a call to Relax's assert_op operation (`assert` is reserved in Python, so the name must be distinct). Parameters ---------- condition: Union[Expr, PrimExpr] The assertion condition. format_args: Optional[Union[Expr, List[Expr]]] Format arguments for

(
    condition: Expr | PrimExpr,
    format_args: Expr | list[Expr] | None = None,
    format: str | Expr = "",
)

def assert_op(
    condition: Expr | PrimExpr,
    format_args: Expr | list[Expr] | None = None,
    format: str | Expr = "",
) -> Expr:
    """
    Create a call to Relax's assert_op operation (`assert` is reserved in Python,
    so the name must be distinct).

    Parameters
    ----------
    condition: Union[Expr, PrimExpr]
        The assertion condition.

    format_args: Optional[Union[Expr, List[Expr]]]
        Format arguments for the error message if the condition fails.

    format: Union[str, Expr]
        The format string or StringImm for the error message.

    Returns
    -------
    result : Expr
        A Call to the Relax assert operation.
    """
    if not isinstance(condition, Expr):
        condition = tvm.relax.PrimValue(condition)

    if format_args is None:
        format_args = []
    elif isinstance(format_args, Expr):
        format_args = [format_args]

    if isinstance(format, str):
        format = StringImm(format)

    return _ffi_api.assert_op(condition, format_args, format)  # type: ignore


def shape_of(expr: Expr) -> Expr: