[tvm] branch ci-docker-staging updated: fix TF tests

[masahi]

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 b6b4d4d6db fix TF tests
b6b4d4d6db is described below

commit b6b4d4d6db8cd4e8f17875e23994ab08761f5298
Author: Masahiro Masuda <masahi...@gmail.com>
AuthorDate: Mon Mar 20 14:18:43 2023 +0900

    fix TF tests
---
 tests/python/frontend/tensorflow/test_forward.py | 6 ++++--
 tests/scripts/request_hook/request_hook.py       | 1 +
 2 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/tests/python/frontend/tensorflow/test_forward.py 
b/tests/python/frontend/tensorflow/test_forward.py
index 1ca0f3faef..a614d05ffc 100644
--- a/tests/python/frontend/tensorflow/test_forward.py
+++ b/tests/python/frontend/tensorflow/test_forward.py
@@ -249,6 +249,8 @@ def compare_tf_with_tvm(
     targets=None,
     ignore_in_shape=False,
     convert_config=None,
+    atol=1e-5,
+    rtol=1e-5,
 ):
     """Generic function to generate and compare tensorflow and TVM output"""
 
@@ -303,7 +305,7 @@ def compare_tf_with_tvm(
             for i, tf_out in enumerate(tf_output):
                 if not isinstance(tf_out, np.ndarray):
                     assert len(tvm_output[i].shape) == 0  # pylint: 
disable=len-as-condition
-                tvm.testing.assert_allclose(tf_out, tvm_output[i], atol=1e-5, 
rtol=1e-5)
+                tvm.testing.assert_allclose(tf_out, tvm_output[i], atol=atol, 
rtol=rtol)
 
         sess.close()
 
@@ -3415,7 +3417,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")
+        compare_tf_with_tvm([image], ["in_data:0"], "crop_and_resize:0", 
atol=1e-4, rtol=1e-4)
 
 
 def test_forward_crop_and_resize():
diff --git a/tests/scripts/request_hook/request_hook.py 
b/tests/scripts/request_hook/request_hook.py
index e1b6cc7d59..323443f129 100644
--- a/tests/scripts/request_hook/request_hook.py
+++ b/tests/scripts/request_hook/request_hook.py
@@ -109,6 +109,7 @@ URL_MAP = {
     
"https://github.com/dmlc/web-data/raw/main/gluoncv/detection/street_small.jpg": 
f"{BASE}/2022-10-05/gluon-small-stree.jpg",
     
"https://github.com/dmlc/web-data/raw/main/tensorflow/models/Custom/placeholder.pb":
 f"{BASE}/dmlc/web-data/raw/main/tensorflow/models/Custom/placeholder.pb",
     
"https://github.com/dmlc/web-data/raw/main/tensorflow/models/InceptionV1/classify_image_graph_def-with_shapes.pb":
 
f"{BASE}/dmlc/web-data/raw/main/tensorflow/models/InceptionV1/classify_image_graph_def-with_shapes.pb",
+    
"https://github.com/dmlc/web-data/raw/main/tensorflow/models/ResnetV2/resnet-20180601_resnet_v2_imagenet-shapes.pb":
 
f"{BASE}/dmlc/web-data/raw/main/tensorflow/models/ResnetV2/resnet-20180601_resnet_v2_imagenet-shapes.pb",
     
"https://github.com/dmlc/web-data/raw/main/tensorflow/models/InceptionV1/elephant-299.jpg":
 
f"{BASE}/dmlc/web-data/raw/main/tensorflow/models/InceptionV1/elephant-299.jpg",
     
"https://github.com/dmlc/web-data/raw/main/tensorflow/models/InceptionV1/imagenet_2012_challenge_label_map_proto.pbtxt":
 
f"{BASE}/dmlc/web-data/raw/main/tensorflow/models/InceptionV1/imagenet_2012_challenge_label_map_proto.pbtxt",
     
"https://github.com/dmlc/web-data/raw/main/tensorflow/models/InceptionV1/imagenet_synset_to_human_label_map.txt":
 
f"{BASE}/dmlc/web-data/raw/main/tensorflow/models/InceptionV1/imagenet_synset_to_human_label_map.txt",