commit 82593deacc7e97b9a667469bcee4392b78a7a768
author wilisa01 <william.isaksson@arm.com> Thu Dec 22 08:00:08 2022 +0000
committer Rickard Bolin <rickard.bolin@arm.com> Fri Jan 20 15:49:01 2023 +0000
tree ad99b40940ebe67e1bbf23491898e77c9984c5bf
parent 3b1578e44b4c6a8c8c9a8e0891d3866a89bd66af

MLBEDSW-6954: Update to TensorFlow 2.11

Updated FlatBuffers autogenerated files to TensorFlow 2.11

Change-Id: Ied60f9fbacdcf91ec8d289cafbde0d88169bb349
Signed-off-by: wilisa01 <william.isaksson@arm.com>
Signed-off-by: Fredrik Svedberg <fredrik.svedberg@arm.com>

ethosu/vela/tflite_mapping.py

diff --git a/ethosu/vela/tflite_mapping.py b/ethosu/vela/tflite_mapping.py
index 8ec0173..11f74ba 100644
--- a/ethosu/vela/tflite_mapping.py
+++ b/ethosu/vela/tflite_mapping.py
@@ -1,4 +1,4 @@
-# SPDX-FileCopyrightText: Copyright 2020-2022 Arm Limited and/or its affiliates <open-source-office@arm.com>
+# SPDX-FileCopyrightText: Copyright 2020-2023 Arm Limited and/or its affiliates <open-source-office@arm.com>
 #
 # SPDX-License-Identifier: Apache-2.0
 #
@@ -162,6 +162,7 @@
 
 
 datatype_map = {
+    TensorType.INT4: DataType.int4,
     TensorType.UINT8: DataType.uint8,
     TensorType.INT8: DataType.int8,
     TensorType.INT16: DataType.int16,
@@ -184,6 +185,7 @@
 datatype_inv_map = inverse_map(datatype_map)
 datatype_inv_map[DataType.quint8] = TensorType.UINT8
 
+datatype_inv_map[DataType.qint4] = TensorType.INT4
 datatype_inv_map[DataType.qint8] = TensorType.INT8
 datatype_inv_map[DataType.qint16] = TensorType.INT16
 datatype_inv_map[DataType.qint32] = TensorType.INT32