From 8dbe0db132cfbb2018f9958aa5749d2d79835f39 Mon Sep 17 00:00:00 2001 From: DvirDukhan Date: Sun, 15 May 2022 15:04:27 +0300 Subject: [PATCH] TF 2.8 --- get_deps.sh | 2 +- tests/flow/tests_dag_errors.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/get_deps.sh b/get_deps.sh index 037988028..cdecbedc6 100755 --- a/get_deps.sh +++ b/get_deps.sh @@ -4,7 +4,7 @@ ORT_VERSION="1.9.0" DLPACK_VERSION="v0.5_RAI" -TF_VERSION="2.6.0" +TF_VERSION="2.8.0" TFLITE_VERSION="2.0.0" PT_VERSION="1.10.1" diff --git a/tests/flow/tests_dag_errors.py b/tests/flow/tests_dag_errors.py index 331510854..d19f86857 100644 --- a/tests/flow/tests_dag_errors.py +++ b/tests/flow/tests_dag_errors.py @@ -282,6 +282,6 @@ def test_dag_error_before_tensorget_op(env): env.assertEqual(type(ret[2]), redis.exceptions.ResponseError) expected_error_msg = 'Incompatible shapes: [2] vs. [2,2,3] \t [[{{node mul}}]]' if DEVICE != 'CPU': - expected_error_msg = 'Invalid argument: required broadcastable shapes' \ + expected_error_msg = 'required broadcastable shapes' \ ' [[{{node mul}}]]' env.assertContains(expected_error_msg, str(ret[2]))