xref: /aosp_15_r20/external/pytorch/tools/onnx/templates/rules.py.in (revision da0073e96a02ea20f0ac840b70461e3646d07c45)
1"""
2${generated_comment}
3"""
4
5import dataclasses
6from typing import Tuple
7
8# flake8: noqa
9from torch.onnx._internal.diagnostics import infra
10
11"""
12${generated_rule_class_comment}
13"""
14
15${rule_classes}
16
17@dataclasses.dataclass
18class _POERules(infra.RuleCollection):
19${rules}
20
21rules = _POERules()
22