saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-30 20:18:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Remove reference to internals in torch.onnx in test (#17550)

Browse source 
- https://github.com/microsoft/onnxruntime/issues/11901
This commit is contained in:
Justin Chu 2023-09-18 18:40:09 -07:00 committed by GitHub
parent f969e7f8d8
commit d350ab31d7
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
onnxruntime/test/python/test_pytorch_export_contrib_ops.py
View file

@ -49,9 +49,7 @@ def ort_test_with_input(ort_sess, input, output, rtol, atol):
# These set of tests verify ONNX model export and compares outputs between
# PyTorch and ORT.
class ONNXExporterTest(unittest.TestCase):
from torch.onnx.symbolic_helper import _export_onnx_opset_version
opset_version = _export_onnx_opset_version
opset_version = 17
keep_initializers_as_inputs = True # For IR version 3 type export.
def setUp(self):