diff options
| author | Charles Xu <charles.xu@arm.com> | 2020-07-02 15:12:40 +0200 |
|---|---|---|
| committer | Fredrik Knutsson <fredrik.knutsson.hunnebo@gmail.com> | 2020-08-05 10:45:33 +0000 |
| commit | 9a03fdff316662be69a1adc4e391e43bc6519b08 (patch) | |
| tree | 6608df66254a762da1660693ee36481e76e97b8c /ethosu/vela/npu_serialisation.py | |
| parent | 0430882a652368d1da1638114d4fbf8c8f7c8d7b (diff) | |
| download | ethos-u-vela-9a03fdff316662be69a1adc4e391e43bc6519b08.tar.gz | |
MLBEDSW-2569:Support 1x1 IFM ResizeBilinear
Signed-off-by: Charles Xu <charles.xu@arm.com>
Change-Id: I44428d77b2e8e44a477e5c4dfe28ab8dd1792838
Diffstat (limited to 'ethosu/vela/npu_serialisation.py')
| -rw-r--r-- | ethosu/vela/npu_serialisation.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/ethosu/vela/npu_serialisation.py b/ethosu/vela/npu_serialisation.py index 2d1c6b10..4b5a888f 100644 --- a/ethosu/vela/npu_serialisation.py +++ b/ethosu/vela/npu_serialisation.py @@ -51,8 +51,9 @@ 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 - end_addr = start_addr + src_tensor.quant_values.size - memory_tensor.values[start_addr:end_addr] = src_tensor.quant_values + values = src_tensor.quant_values.flatten() + end_addr = start_addr + values.size + memory_tensor.values[start_addr:end_addr] = values def serialise_npu_subgraph_into_tensors(nng, sg, arch, scratch_tens, scratch_fast_tens, flash_tens): |