pytorch/torch/onnx
Justin Chu c66e58b7d0 [ONNX] Expose the testing module (#162495)
* Created a new module `torch/onnx/testing.py` that exposes the `assert_onnx_program` function for testing exported ONNX models.
* Updated the ONNX documentation (`docs/source/onnx.md`) to include `onnx_testing` in the list of relevant modules.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/162495
Approved by: https://github.com/titaiwangms, https://github.com/xadupre
2025-09-10 01:40:24 +00:00
..
_internal Improve typing of ONNX decorators with ParamSpec (#162332) 2025-09-07 18:06:03 +00:00
ops Improve typing of ONNX decorators with ParamSpec (#162332) 2025-09-07 18:06:03 +00:00
__init__.py [ONNX] Default to dynamo export (#159646) 2025-09-02 22:45:55 +00:00
_constants.py [ONNX] Set default opset to 20 (#158802) 2025-07-22 19:55:05 +00:00
_flags.py [ONNX] Hide draft export under a flag (#162225) 2025-09-05 19:54:50 +00:00
errors.py [ONNX] Clean up the diagnostics module (#149864) 2025-03-26 05:58:32 +00:00
operators.py remove allow-untyped-defs from torch/onnx/operators.py (#144133) 2025-01-03 20:07:56 +00:00
README.md [ONNX] Refactor torchscript based exporter (#161323) 2025-09-02 16:10:30 +00:00
symbolic_helper.py [ONNX] Refactor torchscript based exporter (#161323) 2025-09-02 16:10:30 +00:00
symbolic_opset7.py [ONNX] Refactor torchscript based exporter (#161323) 2025-09-02 16:10:30 +00:00
symbolic_opset8.py [ONNX] Refactor torchscript based exporter (#161323) 2025-09-02 16:10:30 +00:00
symbolic_opset9.py [ONNX] Refactor torchscript based exporter (#161323) 2025-09-02 16:10:30 +00:00
symbolic_opset10.py [ONNX] Refactor torchscript based exporter (#161323) 2025-09-02 16:10:30 +00:00
symbolic_opset11.py [ONNX] Refactor torchscript based exporter (#161323) 2025-09-02 16:10:30 +00:00
symbolic_opset12.py [ONNX] Refactor torchscript based exporter (#161323) 2025-09-02 16:10:30 +00:00
symbolic_opset13.py [ONNX] Refactor torchscript based exporter (#161323) 2025-09-02 16:10:30 +00:00
symbolic_opset14.py [ONNX] Refactor torchscript based exporter (#161323) 2025-09-02 16:10:30 +00:00
symbolic_opset15.py [ONNX] Refactor torchscript based exporter (#161323) 2025-09-02 16:10:30 +00:00
symbolic_opset16.py [ONNX] Refactor torchscript based exporter (#161323) 2025-09-02 16:10:30 +00:00
symbolic_opset17.py [ONNX] Refactor torchscript based exporter (#161323) 2025-09-02 16:10:30 +00:00
symbolic_opset18.py [ONNX] Refactor torchscript based exporter (#161323) 2025-09-02 16:10:30 +00:00
symbolic_opset19.py [ONNX] Refactor torchscript based exporter (#161323) 2025-09-02 16:10:30 +00:00
symbolic_opset20.py [ONNX] Refactor torchscript based exporter (#161323) 2025-09-02 16:10:30 +00:00
testing.py [ONNX] Expose the testing module (#162495) 2025-09-10 01:40:24 +00:00
utils.py [ONNX] Refactor torchscript based exporter (#161323) 2025-09-02 16:10:30 +00:00
verification.py [ONNX] Refactor torchscript based exporter (#161323) 2025-09-02 16:10:30 +00:00

torch.onnx

Torch->ONNX converter / exporter.