From f7e810a695c1426799d945d61671126543efc123 Mon Sep 17 00:00:00 2001 From: Tim Hall Date: Thu, 25 Jun 2020 15:04:31 +0100 Subject: 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 --- ethosu/vela/tensor.py | 3 +++ 1 file changed, 3 insertions(+) (limited to 'ethosu/vela/tensor.py') 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 -- cgit v1.2.1