diff --git a/tests/python/frontend/onnx/test_forward.py b/tests/python/frontend/onnx/test_forward.py index 91d3911da530..664ed360ceba 100644 --- a/tests/python/frontend/onnx/test_forward.py +++ b/tests/python/frontend/onnx/test_forward.py @@ -1840,7 +1840,16 @@ def verify_reduce_func(func, data, axis, keepdims): model = helper.make_model(graph, producer_name="reduce_test") - verify_with_ort_with_inputs(model, [data], [outshape], opset=11, target=target, dev=dev) + verify_with_ort_with_inputs( + model, + [data], + [outshape], + opset=11, + target=target, + dev=dev, + rtol=1e-4, + atol=1e-4, + ) funcs = [ "ReduceMax",