diff options
author | Louis Verhaard <louis.verhaard@arm.com> | 2020-12-18 14:23:06 +0100 |
---|---|---|
committer | Louis Verhaard <louis.verhaard@arm.com> | 2020-12-22 16:04:38 +0100 |
commit | f4e12beff027e8463aeaed7b75d5e48afb25c8b3 (patch) | |
tree | 6e56a37b7ca30c9f8acdafc08a4627bdadb374d2 /ethosu/vela/tflite_mapping.py | |
parent | f27a8b65f8cff8fc52db8e39a6eb8f78b6616c6b (diff) | |
download | ethos-u-vela-f4e12beff027e8463aeaed7b75d5e48afb25c8b3.tar.gz |
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
Diffstat (limited to 'ethosu/vela/tflite_mapping.py')
-rw-r--r-- | ethosu/vela/tflite_mapping.py | 2 |
1 files changed, 1 insertions, 1 deletions
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, } |