Function function

python/tvm/relax/script/builder/ir.py:261–278

Start a function frame. Parameters ---------- is_pure: bool Whether the function is annotated as pure. is_private : bool Whether the function is annotated as private. Returns ------- frame: FunctionFrame The constructed function frame.

(is_pure: bool = True, is_private: bool = False)

def function(is_pure: bool = True, is_private: bool = False) -> frame.FunctionFrame:
    """Start a function frame.
    Parameters
    ----------
    is_pure: bool
        Whether the function is annotated as pure.

    is_private : bool
        Whether the function is annotated as private.

    Returns
    -------
    frame: FunctionFrame
        The constructed function frame.
    """
    return _ffi_api.Function(  # type: ignore[attr-defined]  # pylint: disable=no-member
        is_pure, is_private
    )


def arg(name: py_str, struct_info: StructInfo) -> Var: