diff options
| author | Fredrik Svedberg <fredrik.svedberg@arm.com> | 2020-08-27 15:51:50 +0200 |
|---|---|---|
| committer | tim.hall <tim.hall@arm.com> | 2020-08-27 15:32:40 +0000 |
| commit | bb1a92ab226cac13a138ab6570fe8a060d1dafa9 (patch) | |
| tree | 279fd31227b47ee4856abe9b54c3a32511a7fc21 /ethosu/vela/npu_serialisation.py | |
| parent | 880e73543120648f08886365a45e8b2ce32d5ff1 (diff) | |
| download | ethos-u-vela-bb1a92ab226cac13a138ab6570fe8a060d1dafa9.tar.gz | |
[MLBEDSW-2857] Fix serialisation of non byte tensors
Fixed serialisation of scalar ifm tensors with values larger than
byte sized.
Signed-off-by: Fredrik Svedberg <fredrik.svedberg@arm.com>
Change-Id: I2714398db91b83f24e5271c1d5de1c0e8211f9ab
Diffstat (limited to 'ethosu/vela/npu_serialisation.py')
| -rw-r--r-- | ethosu/vela/npu_serialisation.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/ethosu/vela/npu_serialisation.py b/ethosu/vela/npu_serialisation.py index c6b0d877..6277a6dc 100644 --- a/ethosu/vela/npu_serialisation.py +++ b/ethosu/vela/npu_serialisation.py @@ -52,6 +52,8 @@ 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() + if src_tensor.dtype.size_in_bytes() > 1: + values = np.frombuffer(values.tobytes(), dtype=np.uint8) end_addr = start_addr + values.size memory_tensor.values[start_addr:end_addr] = values |