From b2f4abbbd6f26740e23b0d3e0e34152b9b4db63b Mon Sep 17 00:00:00 2001 From: Joseph Tummon Date: Mon, 24 Jul 2023 14:19:36 +0100 Subject: MLIA-757: Update Vela and TensorFlow dependencies Ignoring new 'Conv_hwcn_Weights' layer that is named in TensorFlow 2.11.1 and used as buffer for float32 only. Signed-off-by: Joseph Tummon Change-Id: Iaa579c76013b1e0dc505466c46f5864a96af8c6d --- setup.cfg | 4 ++-- src/mlia/nn/tensorflow/tflite_metrics.py | 2 ++ 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/setup.cfg b/setup.cfg index bae275b..aa513f5 100644 --- a/setup.cfg +++ b/setup.cfg @@ -29,9 +29,9 @@ package_dir = = src packages = find_namespace: install_requires = - tensorflow~=2.10.1 + tensorflow~=2.11.1 tensorflow-model-optimization~=0.7.3 - ethos-u-vela~=3.7.0 + ethos-u-vela~=3.8.0 requests~=2.28.1 rich~=12.6.0 tomli~=2.0.1 ; python_version<"3.11" diff --git a/src/mlia/nn/tensorflow/tflite_metrics.py b/src/mlia/nn/tensorflow/tflite_metrics.py index d7ae2a4..4c71dc4 100644 --- a/src/mlia/nn/tensorflow/tflite_metrics.py +++ b/src/mlia/nn/tensorflow/tflite_metrics.py @@ -29,6 +29,8 @@ DEFAULT_IGNORE_LIST = [ "flatten", "StatefulPartitionedCall", "bias", + # Buffer layer from TensorFlow 2.11 (previously unamed) + "Conv_hwcn_weights", ] -- cgit v1.2.1