mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
d3ef5c3fa3
- Move definitions in `__init__` to internal classes and expose them by importing to init (prevent circular dependencies): https://github.com/pytorch/pytorch/wiki/torch.onnx-Namespacing - Context classes and enums are moved to `_exporter_states.py` - Exceptions are moved to `errors.py` - Define `__all__` for torch.onnx. https://github.com/pytorch/pytorch/wiki/Public-API-definition-and-documentation - Moved `utils.__IN_ONNX_EXPORT` to `GLOBALS.in_onnx_export` - Deprecated `torch.onnx._export` Precedes #78231 Using this as an aid for finding public functions: ```python list(filter(lambda x: not x.startswith("_"), torch.onnx.utils.__dict__.keys())) ``` Pull Request resolved: https://github.com/pytorch/pytorch/pull/78446 Approved by: https://github.com/BowenBao
10 lines
325 B
Python
10 lines
325 B
Python
"""Constant values used in ONNX."""
|
|
|
|
ONNX_ARCHIVE_MODEL_PROTO_NAME = "__MODEL_PROTO"
|
|
onnx_default_opset = 13
|
|
onnx_main_opset = 16
|
|
onnx_stable_opsets = tuple(range(7, onnx_main_opset))
|
|
onnx_constant_folding_opsets = tuple(range(9, onnx_main_opset + 1))
|
|
|
|
PYTORCH_GITHUB_ISSUES_URL = "https://github.com/pytorch/pytorch/issues"
|