Class TupleGetItemPattern

python/tvm/relax/dpl/pattern.py:495–513

Get index-th item from a TuplePattern. Parameters ---------- tuple_value: tvm.relax.dpl.DFPattern The input tuple expression. index: Optional[int] The index to match; Default (None) to match a TupleGetItem with any index.

@register_df_node
class TupleGetItemPattern(DFPattern):
    """Get index-th item from a TuplePattern.

    Parameters
    ----------
    tuple_value: tvm.relax.dpl.DFPattern
        The input tuple expression.

    index: Optional[int]
        The index to match; Default (None) to match a TupleGetItem with any index.
    """

    def __init__(self, tuple_value: "DFPattern", index: int | None = None):
        match_index = index if index is not None else -1
        self.__init_handle_by_constructor__(
            ffi.TupleGetItemPattern,
            tuple_value,
            match_index,  # type: ignore
        )


@register_df_node