From ea4994517f84475dab99f7dc3226b937da7f6b96 Mon Sep 17 00:00:00 2001 From: Tim Hall Date: Tue, 7 Jul 2020 12:04:37 +0100 Subject: MLBEDSW-2615: MLCE: Ethos-U55 Flash usage increases due to Yoda Beta - Fixed bug with the size of the scale and bias tensor Signed-off-by: Tim Hall Change-Id: I4d267d4c918a5c834ebdff82de4f021717e95203 --- ethosu/vela/weight_compressor.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) (limited to 'ethosu') diff --git a/ethosu/vela/weight_compressor.py b/ethosu/vela/weight_compressor.py index 2554b7c8..df7ff751 100644 --- a/ethosu/vela/weight_compressor.py +++ b/ethosu/vela/weight_compressor.py @@ -424,7 +424,8 @@ def calc_scales_and_pack_biases(tens, arch, ofm_depth_step, rescale_for_faf=Fals tens.compressed_values.append(stream) tens.compressed_values_substream_offsets.append(substream_offsets) - tens.storage_shape = [total_elements * tens.element_size_bytes] + tens.storage_shape = [total_elements] + def update_pass_weight_and_scale_tensors(nng, arch): -- cgit v1.2.1