Class ShapePattern

python/tvm/relax/dpl/pattern.py:609–622

A pattern that matches another pattern with a certain tensor shape Parameters ---------- pattern: tvm.relax.dpl.DFPattern The input pattern that needs type annotation. shape: List[tvm.ir.PrimExpr] The shape to match.

@register_df_node
class ShapePattern(DFPattern):
    """A pattern that matches another pattern with a certain tensor shape

    Parameters
    ----------
    pattern: tvm.relax.dpl.DFPattern
        The input pattern that needs type annotation.

    shape: List[tvm.ir.PrimExpr]
        The shape to match.
    """

    def __init__(self, pattern: "DFPattern", shape: list[tvm.ir.PrimExpr]):
        self.__init_handle_by_constructor__(ffi.ShapePattern, pattern, shape)  # type: ignore


@register_df_node