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_reader.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_reader.py')
-rw-r--r-- | ethosu/vela/tflite_reader.py | 14 |
1 files changed, 9 insertions, 5 deletions
diff --git a/ethosu/vela/tflite_reader.py b/ethosu/vela/tflite_reader.py index 21ff8873..45397c2b 100644 --- a/ethosu/vela/tflite_reader.py +++ b/ethosu/vela/tflite_reader.py @@ -129,11 +129,15 @@ class TFLiteSubgraph: tens.values = None buf = self.graph.buffers[tens_data.Buffer()] - if buf is not None and dtype != DataType.string: - tens.values = np.array(buf.view(datatype_map_numpy[tens_dtype]).reshape(shape)) - if tens.quantization is not None: - tens.quant_values = tens.values - tens.values = tens.quantization.dequantize(tens.quant_values) + if buf is not None: + np_dtype = datatype_map_numpy[tens_dtype] + if dtype == DataType.string: + tens.values = np.array(buf.view(np_dtype)) + else: + tens.values = np.array(buf.view(np_dtype).reshape(shape)) + if tens.quantization is not None: + tens.quant_values = tens.values + tens.values = tens.quantization.dequantize(tens.quant_values) return tens def parse_operator(self, op_index, op_data): |