Class FlattenNode

backends/example/example_operators/flatten.py:35–40

@dataclass
class FlattenNode(OpBase):
    def __init__(self):
        super().__init__(
            pattern=(torch.flatten,),
            annotate_handle=_annotate_flatten,
        )