xref: /aosp_15_r20/external/pytorch/tools/onnx/templates/rules.py.in (revision da0073e96a02ea20f0ac840b70461e3646d07c45)

"""
${generated_comment}
"""

import dataclasses
from typing import Tuple

# flake8: noqa
from torch.onnx._internal.diagnostics import infra

"""
${generated_rule_class_comment}
"""

${rule_classes}

@dataclasses.dataclass
class _POERules(infra.RuleCollection):
${rules}

rules = _POERules()