Class Call

python/tvm/relax/expr.py:534–584

Function call node in Relax. Call node corresponds the operator application node in computational graph terminology. Parameters ---------- op: tvm.ir.Op or any tvm.relax.Expr with function type. The operation to be called. args: Union[List[Expr], typing.Tuple[Expr,

@tvm_ffi.register_object("relax.expr.Call")
class Call(ExprWithOp):
    """Function call node in Relax.

    Call node corresponds the operator application node
    in computational graph terminology.

    Parameters
    ----------
    op: tvm.ir.Op or any tvm.relax.Expr with function type.
        The operation to be called.

    args: Union[List[Expr], typing.Tuple[Expr, ...]]
        The arguments to the call.

    attrs: Optional[tvm.ir.Attrs]
        Attributes to the call, can be None

    sinfo_args: Optional[Union[List[StructInfo], typing.Tuple[StructInfo, ...]]]
        The structure info arguments of a CallNode.
        sinfo_args is designed to be non-empty only for intrinsic op (e.g.,
        call_tir, call_builtin_with_ctx, etc.) and calls to ExternFuncs, with the main
        usage of structure info inference.

    span: Optional[Span]
        Span that points to original source code
    """

    op: Expr
    args: list[Expr]
    attrs: tvm.ir.Attrs
    sinfo_args: list[StructInfo]
    span: Span | None

    def __init__(
        self,
        op: Expr | tvm.ir.Op,
        args: list[Expr] | tuple[Expr, ...],
        attrs: tvm.ir.Attrs | None = None,
        sinfo_args: list[StructInfo] | tuple[StructInfo, ...] | None = None,
        span: Span | None = None,
    ):
        if not sinfo_args:
            sinfo_args = []
        self.__init_handle_by_constructor__(
            _ffi_api.Call,
            op,
            args,
            attrs,
            sinfo_args,
            span,  # type: ignore
        )


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