mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 12:54:11 +08:00
b90aa18569
Re-land of https://github.com/pytorch/pytorch/pull/125242 Pull Request resolved: https://github.com/pytorch/pytorch/pull/127034 Approved by: https://github.com/malfet
16 lines
241 B
Python
16 lines
241 B
Python
from dataclasses import dataclass
|
|
from typing import List
|
|
|
|
from torch._export.serde.schema import Node
|
|
|
|
|
|
@dataclass
|
|
class ExternKernelNode:
|
|
name: str
|
|
node: Node
|
|
|
|
|
|
@dataclass
|
|
class ExternKernelNodes:
|
|
nodes: List[ExternKernelNode]
|