diff options
author | Diqing Zhong <diqing.zhong@arm.com> | 2020-10-30 15:10:46 +0100 |
---|---|---|
committer | Diqing Zhong <diqing.zhong@arm.com> | 2020-10-30 15:11:59 +0100 |
commit | 7e1d1d181d5334bf50fc2e1f39f11c20a22fc341 (patch) | |
tree | 86081ed2bc00f2a8106a065e681a5f46a5baea3e | |
parent | a151f597bef24e0d8b51dbe833338057e8bcbc92 (diff) | |
download | ethos-u-vela-7e1d1d181d5334bf50fc2e1f39f11c20a22fc341.tar.gz |
Vela: Fix wrong bandwidth
- copy bandwidth compression rate when weight tensor is cloned
Signed-off-by: Diqing Zhong <diqing.zhong@arm.com>
Change-Id: I41c4c1f7001e8dc12af35695f5f5d02815e28351
-rw-r--r-- | ethosu/vela/tensor.py | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/ethosu/vela/tensor.py b/ethosu/vela/tensor.py index 4110fa67..0fcf9a97 100644 --- a/ethosu/vela/tensor.py +++ b/ethosu/vela/tensor.py @@ -440,6 +440,7 @@ class Tensor: self.weight_transpose_depthwise = src_tens.weight_transpose_depthwise self.compression_scale_for_worst_weight_stream = src_tens.compression_scale_for_worst_weight_stream self.storage_compression_scale = src_tens.storage_compression_scale + self.bandwidth_compression_scale = src_tens.bandwidth_compression_scale self.block_traversal = src_tens.block_traversal self.weight_compression_config = src_tens.weight_compression_config self.value_id = src_tens.value_id |