diff options
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): |