From 0f98b361288c71fca327969346db32de098c797b Mon Sep 17 00:00:00 2001 From: Fredrik Svedberg Date: Tue, 29 Sep 2020 10:00:39 +0200 Subject: [MLBEDSW-2802] Fix 5D tensor crash Fixed crash in networks with 5D tensors. Fixed crash for (int32) tensors without quantization. Added validity checks for concatenation. Moved unfusing of activation function from tflite_reader to graph_optimiser. Signed-off-by: Fredrik Svedberg Change-Id: Ib9ba8891dc95ef5491e15d0feedef44331a26393 --- ethosu/vela/npu_serialisation.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'ethosu/vela/npu_serialisation.py') diff --git a/ethosu/vela/npu_serialisation.py b/ethosu/vela/npu_serialisation.py index 6277a6dc..430db585 100644 --- a/ethosu/vela/npu_serialisation.py +++ b/ethosu/vela/npu_serialisation.py @@ -51,7 +51,7 @@ def copy_compressed_values_to_memory_tensor(memory_tensor, src_tensor): def copy_ifm_values_to_memory_tensor(memory_tensor, src_tensor): start_addr = src_tensor.address - values = src_tensor.quant_values.flatten() + values = src_tensor.quant_values.flatten() if src_tensor.quant_values is not None else src_tensor.values.flatten() if src_tensor.dtype.size_in_bytes() > 1: values = np.frombuffer(values.tobytes(), dtype=np.uint8) end_addr = start_addr + values.size -- cgit v1.2.1