mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
8d8fb7efe7
Pull Request resolved: https://github.com/pytorch/pytorch/pull/94565 Approved by: https://github.com/abock
22 lines
292 B
Python
22 lines
292 B
Python
"""
|
|
${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()
|