Function MergeCompositeFunctions

python/tvm/relax/transform/transform.py:965–975

Group one or multiple composite functions created by FuseOpsByPattern into a new function. The new function will be annotated with "Codegen" and "global_symbol" attributes, and it is intented to be offloaded to an external backend. Returns ------- ret : tvm.transform.Pass

()

def MergeCompositeFunctions() -> tvm.ir.transform.Pass:
    """Group one or multiple composite functions created by FuseOpsByPattern into a new function.
    The new function will be annotated with "Codegen" and "global_symbol" attributes, and it
    is intented to be offloaded to an external backend.

    Returns
    -------
    ret : tvm.transform.Pass
        The registered pass for merging composite functions.
    """
    return _ffi_api.MergeCompositeFunctions()  # type: ignore


def AttachAttrLayoutFreeBuffers() -> tvm.ir.transform.Pass: