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

Restore tolerance fix and make test_gru_seq_length_cpu run (#595)

Browse source
This commit is contained in:
Dmitri Smirnov 2019-03-11 16:28:30 -07:00 committed by GitHub
parent 02ad7daa8b
commit 59cfafbb9e
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
onnxruntime/test/python/onnx_backend_test_series.py
View file

@ -11,10 +11,10 @@ import onnxruntime.backend as c2
pytest_plugins = 'onnx.backend.test.report',
class OnnxruntimeBackendTest(onnx.backend.test.BackendTest):
class OrtBackendTest(onnx.backend.test.BackendTest):
def __init__(self, backend, parent_module=None):
onnx.backend.test.BackendTest.__init__(self, backend, parent_module)
super(OrtBackendTest, self).__init__(backend, parent_module)
@classmethod
def assert_similar_outputs(cls, ref_outputs, outputs, rtol, atol):
@ -27,18 +27,16 @@ class OnnxruntimeBackendTest(onnx.backend.test.BackendTest):
np.testing.assert_allclose(
ref_outputs[i],
outputs[i],
rtol=rtol,
atol=atol)
rtol=1e-3,
atol=1e-5)
backend_test = OnnxruntimeBackendTest(c2, __name__)
backend_test = OrtBackendTest(c2, __name__)
# Type not supported
backend_test.exclude(r'(FLOAT16)')
backend_test.exclude(r'('
'^test_cast_DOUBLE_to_FLOAT_cpu.*'
'|^test_gru_seq_length_cpu.*'
'|^test_cast_FLOAT_to_DOUBLE_cpu.*'
'|^test_cast_FLOAT_to_STRING_cpu.*'
'|^test_cast_STRING_to_FLOAT_cpu.*'