This is an automated email from the ASF dual-hosted git repository.
masahi pushed a commit to branch ci-docker-staging
in repository https://gitbox.apache.org/repos/asf/tvm.git
The following commit(s) were added to refs/heads/ci-docker-staging by this push:
new 8ab12931fc more tf fix
8ab12931fc is described below
commit 8ab12931fc1e5d8a5735e4391754cc4a20d3ba60
Author: Masahiro Masuda <[email protected]>
AuthorDate: Mon Mar 20 17:32:02 2023 +0900
more tf fix
---
tests/python/frontend/tensorflow/test_forward.py | 6 +++++-
1 file changed, 5 insertions(+), 1 deletion(-)
diff --git a/tests/python/frontend/tensorflow/test_forward.py
b/tests/python/frontend/tensorflow/test_forward.py
index a614d05ffc..f6dd933553 100644
--- a/tests/python/frontend/tensorflow/test_forward.py
+++ b/tests/python/frontend/tensorflow/test_forward.py
@@ -3403,6 +3403,8 @@ def _test_forward_crop_and_resize(
extrapolation_value=0.0,
method="bilinear",
dtype="float32",
+ atol=1e-4,
+ rtol=1e-4,
):
image = np.random.uniform(0, 10, size=img_shape).astype(dtype)
tf.reset_default_graph()
@@ -3417,7 +3419,7 @@ def _test_forward_crop_and_resize(
extrapolation_value=extrapolation_value,
name="crop_and_resize",
)
- compare_tf_with_tvm([image], ["in_data:0"], "crop_and_resize:0",
atol=1e-4, rtol=1e-4)
+ compare_tf_with_tvm([image], ["in_data:0"], "crop_and_resize:0",
atol=atol, rtol=rtol)
def test_forward_crop_and_resize():
@@ -3454,6 +3456,8 @@ def test_forward_crop_and_resize():
crop_size=[58, 58],
extrapolation_value=0.2,
method="nearest",
+ atol=1e-3,
+ rtol=1e-3,
)