diff options
author | Joseph Tummon <Joseph.Tummon@arm.com> | 2023-07-24 14:19:36 +0100 |
---|---|---|
committer | Joseph Tummon <Joseph.Tummon@arm.com> | 2023-08-22 12:11:58 +0100 |
commit | b2f4abbbd6f26740e23b0d3e0e34152b9b4db63b (patch) | |
tree | 9effd57a78885a0c12f9d3055a4b0af1d02100b5 | |
parent | 87647b7bf1ee89b20aa87768ec8c82f329082db0 (diff) | |
download | mlia-b2f4abbbd6f26740e23b0d3e0e34152b9b4db63b.tar.gz |
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 <Joseph.Tummon@arm.com>
Change-Id: Iaa579c76013b1e0dc505466c46f5864a96af8c6d
-rw-r--r-- | setup.cfg | 4 | ||||
-rw-r--r-- | src/mlia/nn/tensorflow/tflite_metrics.py | 2 |
2 files changed, 4 insertions, 2 deletions
@@ -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", ] |