mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 12:54:11 +08:00
c66e58b7d0
* 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
9 lines
219 B
Python
9 lines
219 B
Python
"""Utilities to aid in testing exported ONNX models."""
|
|
|
|
__all__ = ["assert_onnx_program"]
|
|
|
|
from torch.onnx._internal.exporter._testing import assert_onnx_program
|
|
|
|
|
|
assert_onnx_program.__module__ = "torch.onnx.testing"
|