1from dataclasses import dataclass 2from typing import List 3 4from torch._export.serde.schema import Node 5 6 7@dataclass 8class ExternKernelNode: 9 name: str 10 node: Node 11 12 13@dataclass 14class ExternKernelNodes: 15 nodes: List[ExternKernelNode] 16