From f4e12beff027e8463aeaed7b75d5e48afb25c8b3 Mon Sep 17 00:00:00 2001
From: Louis Verhaard <louis.verhaard@arm.com>
Date: Fri, 18 Dec 2020 14:23:06 +0100
Subject: MLBEDSW-3561: Support const STRING input tensors

Added handling of input tensors with constant string data.

Signed-off-by: Louis Verhaard <louis.verhaard@arm.com>
Change-Id: Ieb5164a9d56d580ad08ea834bf2cbb7288cd9539
---
 ethosu/vela/tflite_mapping.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'ethosu/vela/tflite_mapping.py')

diff --git a/ethosu/vela/tflite_mapping.py b/ethosu/vela/tflite_mapping.py
index cc6053c0..6fcbb95b 100644
--- a/ethosu/vela/tflite_mapping.py
+++ b/ethosu/vela/tflite_mapping.py
@@ -172,7 +172,7 @@ datatype_map_numpy = {
     TensorType.BOOL: np.bool,
     TensorType.COMPLEX64: np.complex64,
     TensorType.COMPLEX128: np.complex128,
-    TensorType.STRING: np.dtype("S1"),
+    TensorType.STRING: np.uint8,
 }
 
 
-- 
cgit v1.2.1