mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00

Files

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

151 B

Raw Permalink Blame History

torch.onnx.testing

.. automodule:: torch.onnx.testing

.. autofunction:: torch.onnx.testing.assert_onnx_program

151 B Raw Permalink Blame History

torch.onnx.testing

151 B

Raw Permalink Blame History