diff options
| author | Rickard Bolin <rickard.bolin@arm.com> | 2023-04-24 10:39:41 +0000 |
|---|---|---|
| committer | Rickard Bolin <rickard.bolin@arm.com> | 2023-04-25 11:32:40 +0000 |
| commit | 3fd5d839a29e8f0a00638e1c6f86695a5d30d41a (patch) | |
| tree | 16fb2f818073b09b16c8ec610dc072b856f1fdd0 /ethosu/vela/tflite_mapping.py | |
| parent | c4268bf407048c7899c8501dd1223a777f8c4963 (diff) | |
| download | ethos-u-vela-3fd5d839a29e8f0a00638e1c6f86695a5d30d41a.tar.gz | |
MLBEDSW-6954: Update to TensorFlow 2.11
Updated FlatBuffers autogenerated files to TensorFlow 2.11
Change-Id: Ia39d30b06e9a37c9ab119d501ebf442f32167afe
Signed-off-by: Rickard Bolin <rickard.bolin@arm.com>
Diffstat (limited to 'ethosu/vela/tflite_mapping.py')
| -rw-r--r-- | ethosu/vela/tflite_mapping.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/ethosu/vela/tflite_mapping.py b/ethosu/vela/tflite_mapping.py index 14052ce5..bb45a7f2 100644 --- a/ethosu/vela/tflite_mapping.py +++ b/ethosu/vela/tflite_mapping.py @@ -162,6 +162,7 @@ def inverse_map(map): datatype_map = { + TensorType.INT4: DataType.int4, TensorType.UINT8: DataType.uint8, TensorType.INT8: DataType.int8, TensorType.INT16: DataType.int16, @@ -184,6 +185,7 @@ datatype_map = { 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 |