diff options
| author | Tim Hall <tim.hall@arm.com> | 2020-06-25 15:04:31 +0100 |
|---|---|---|
| committer | Tim Hall <tim.hall@arm.com> | 2020-06-25 15:07:58 +0100 |
| commit | f7e810a695c1426799d945d61671126543efc123 (patch) | |
| tree | dad3bfd9120bdcd92d3fe216a01a4e798276a080 /ethosu/vela/tensor.py | |
| parent | eca2e95e1fea150d8a942f8b5f0a4d9d7aefebc1 (diff) | |
| download | ethos-u-vela-f7e810a695c1426799d945d61671126543efc123.tar.gz | |
vela: MLBEDSW-828 weight/scale stream interleaving
- Multicore weight and scale stream interleaving for
multicore hardware architecture.
Change-Id: Ic82850463391c629d90d08c26cf0c48dd438286d
Signed-off-by: Tim Hall <tim.hall@arm.com>
Diffstat (limited to 'ethosu/vela/tensor.py')
| -rw-r--r-- | ethosu/vela/tensor.py | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/ethosu/vela/tensor.py b/ethosu/vela/tensor.py index 3990164d..eda21c9c 100644 --- a/ethosu/vela/tensor.py +++ b/ethosu/vela/tensor.py @@ -229,6 +229,7 @@ class Tensor: "values", "quant_values", "compressed_values", + "compressed_values_substream_offsets", "mem_area", "mem_type", "format", @@ -273,6 +274,7 @@ class Tensor: self.values = None self.quant_values = None self.compressed_values = None + self.compressed_values_substream_offsets = None self.mem_area = MemArea.Unknown self.mem_type = MemType.Unknown self.format = TensorFormat.Unknown @@ -342,6 +344,7 @@ class Tensor: def copy_compressed_weight_info(self, src_tens): # Copies compressed values + all related weight compression info from the given tensor self.compressed_values = src_tens.compressed_values + self.compressed_values_substream_offsets = src_tens.compressed_values_substream_offsets self.storage_shape = src_tens.storage_shape self.brick_size = src_tens.brick_size self.weight_compression_scales = src_tens.weight_compression_scales |