Class FunctionPattern

python/tvm/relax/dpl/pattern.py:428–446

A pattern matching a function node in Relax. Parameters ---------- params: List[tvm.relax.dpl.DFPattern] The parameters to the Function or None to match any parameters. body: tvm.relax.dpl.DFPattern The body fo the Function

@register_df_node
class FunctionPattern(DFPattern):
    """A pattern matching a function node in Relax.

    Parameters
    ----------
    params: List[tvm.relax.dpl.DFPattern]
        The parameters to the Function or None to match any parameters.

    body: tvm.relax.dpl.DFPattern
        The body fo the Function

    """

    def __init__(
        self,
        params: list["DFPattern"],
        body: "DFPattern",
    ):
        self.__init_handle_by_constructor__(ffi.FunctionPattern, params, body)  # type: ignore


@register_df_node