diff options
| author | Dwight Lidman <dwight.lidman@arm.com> | 2020-11-05 15:56:08 +0100 |
|---|---|---|
| committer | patrik.gustavsson <patrik.gustavsson@arm.com> | 2020-11-16 09:10:54 +0000 |
| commit | e05de456770926fdc057478739d1b96b7f651756 (patch) | |
| tree | 784fb9792a959a97c897ece67b23787abbcfcae1 /ethosu/vela/tflite_mapping.py | |
| parent | e8a5a78dd16ec979c7a7bb1f5bd87e9b2909c32d (diff) | |
| download | ethos-u-vela-e05de456770926fdc057478739d1b96b7f651756.tar.gz | |
MLBEDSW-3301: Vela fails ungracefully when reading string buffers
When encountering a sparse string buffer, Vela fails
both due to missing a mapping for a Numpy string type
and also for not being able to read sparse buffers.
The failing line is attempting to reshape a [100]
buffer into a [3, 5] tensor which does not work due
to Vela treating the buffer as non-sparse.
The solution here is to simply not do the reshape
for string buffers (which all appear to be sparse)
since it is not something that will be supported in
the future anyway.
The related operator can then be pushed to the CPU
as expected.
Signed-off-by: Dwight Lidman <dwight.lidman@arm.com>
Change-Id: Iea0af6cd60a691f975209014b6aa098dde8d6a4b
Diffstat (limited to 'ethosu/vela/tflite_mapping.py')
| -rw-r--r-- | ethosu/vela/tflite_mapping.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/ethosu/vela/tflite_mapping.py b/ethosu/vela/tflite_mapping.py index bbd44b0b..44ecedcc 100644 --- a/ethosu/vela/tflite_mapping.py +++ b/ethosu/vela/tflite_mapping.py @@ -171,6 +171,7 @@ datatype_map_numpy = { TensorType.BOOL: np.bool, TensorType.COMPLEX64: np.complex64, TensorType.COMPLEX128: np.complex128, + TensorType.STRING: np.dtype("S1"), } |