natas_hw
/
openvino
0
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[TF FE] Support TensorFlow 2.16 (#23562) 

**Details:** Support TensorFlow 2.16

**Ticket:** TBD

---------

Signed-off-by: Kazantsev, Roman <roman.kazantsev@intel.com>
This commit is contained in:
Roman Kazantsev 2024-03-22 00:39:39 +04:00 committed by GitHub
parent 778f280e3d
commit 1ab9bedd81
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
6 changed files with 11 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
.github/workflows/job_tensorflow_models_tests.yml vendored
View File

@ -114,7 +114,12 @@ jobs:
- name: TensorFlow Models Tests - TF FE
run: |
export PYTHONPATH=${MODEL_HUB_TESTS_INSTALL_DIR}:$PYTHONPATH
python3 -m pytest ${MODEL_HUB_TESTS_INSTALL_DIR}/tensorflow/ -m ${{ inputs.model_scope }} --html=${INSTALL_TEST_DIR}/TEST-tf_fe_models_${{ inputs.model_scope }}.html --self-contained-html -v
python3 -m pytest ${MODEL_HUB_TESTS_INSTALL_DIR}/tensorflow/test_tf_convert_model.py -m ${{ inputs.model_scope }} \
--html=${INSTALL_TEST_DIR}/TEST-tf_fe_models_${{ inputs.model_scope }}.html --self-contained-html -v
# decouple notebook tests due to GitHub issue in tensorflow_hub https://github.com/tensorflow/hub/issues/903
# and use WA to switch to (legacy) Keras 2
TF_USE_LEGACY_KERAS=1 python3 -m pytest ${MODEL_HUB_TESTS_INSTALL_DIR}/tensorflow/test_tf_hub_api_notebooks.py -m ${{ inputs.model_scope }} \
--html=${INSTALL_TEST_DIR}/TEST-tf_fe_models_notebooks_${{ inputs.model_scope }}.html --self-contained-html -v
env:
TEST_DEVICE: CPU

2
src/bindings/python/constraints.txt
View File

@ -19,7 +19,7 @@ patchelf<=0.17.2.1
# Frontends
docopt~=0.6.2
paddlepaddle==2.5.2
tensorflow>=1.15.5,<2.16.0
tensorflow>=1.15.5,<2.17.0
six~=1.16.0
protobuf>=3.18.1,<4.0.0
onnx==1.15.0

2
tests/constraints.txt
View File

@ -13,7 +13,7 @@ sympy>=1.10
wheel>=0.38.1
defusedxml>=0.7.1
fastjsonschema~=2.17.1
tensorflow>=2.5,<2.16.0
tensorflow>=2.5,<2.17.0
test-generator==0.1.2
requests>=2.25.1
opencv-python>=4.5

1
tests/model_hub_tests/tensorflow/test_tf_hub_api_notebooks.py
View File

@ -5,6 +5,7 @@ import pytest
import tensorflow as tf
import tensorflow_hub as hub
from models_hub_common.test_convert_model import TestConvertModel
from utils import get_input_info

2
tools/mo/requirements_tf.txt
View File

@ -1,5 +1,5 @@
-c ../constraints.txt
tensorflow>=1.15.5,<2.16.0
tensorflow>=1.15.5,<2.17.0
numpy>=1.16.6,<1.26
networkx
defusedxml

2
tools/mo/requirements_tf2.txt
View File

@ -1,5 +1,5 @@
-c ../constraints.txt
tensorflow>=2.5,<2.16.0
tensorflow>=2.5,<2.17.0
numpy>=1.16.6,<1.26
networkx
defusedxml