Function _wrap_inline_arg_tuple

python/tvm/relax/op/base.py:70–92

Helper function to wrap argument tuple Normalize the arguments provided the functions that accept a tuple of arguments, and require the tuple of arguments to be written in-line. If the arguments provided are a single relax expression, and are not a reference to a relax tuple, then

(args)

def _wrap_inline_arg_tuple(args) -> Expr:
    """Helper function to wrap argument tuple

    Normalize the arguments provided the functions that accept a tuple
    of arguments, and require the tuple of arguments to be written
    in-line.  If the arguments provided are a single relax expression,
    and are not a reference to a relax tuple, then wrap them into an
    in-line relax Tuple.

    """
    if isinstance(args, tuple | list):
        return tvm.relax.Tuple([convert_to_expr(a) for a in args])
    elif (
        isinstance(args, Expr)
        and not isinstance(args, tvm.relax.Tuple)
        and (
            args.struct_info_ is None
            or not isinstance(args.struct_info_, tvm.relax.TupleStructInfo)
        )
    ):
        return tvm.relax.Tuple([args])
    else:
        return args


def call_tir(