diff options
author | Patrik Gustavsson <patrik.gustavsson@arm.com> | 2020-08-24 16:26:11 +0200 |
---|---|---|
committer | tim.hall <tim.hall@arm.com> | 2020-08-26 09:37:01 +0000 |
commit | 90831bc18d45008b703e59aad0594026beb7da82 (patch) | |
tree | 6b647996160ba8ecd037da1e15c91f55d4b0f665 /ethosu/vela/tensor.py | |
parent | d47cc2777f3a3f371958c30a3c1880f692b0b0a2 (diff) | |
download | ethos-u-vela-90831bc18d45008b703e59aad0594026beb7da82.tar.gz |
MLBED-2822 Added CLI-opt for weight size est.
Added --weight-estimation-scaling, which enables
additional scaling of weight compression scale estimate.
Signed-off-by: Patrik Gustavsson <patrik.gustavsson@arm.com>
Change-Id: Idcda41257f44901d3a3f345341e07fb1ae8585a9
Diffstat (limited to 'ethosu/vela/tensor.py')
-rw-r--r-- | ethosu/vela/tensor.py | 13 |
1 files changed, 9 insertions, 4 deletions
diff --git a/ethosu/vela/tensor.py b/ethosu/vela/tensor.py index d4f6a409..3ad9b253 100644 --- a/ethosu/vela/tensor.py +++ b/ethosu/vela/tensor.py @@ -439,20 +439,25 @@ class Tensor: def has_fully_defined_shape(self): return shape_fully_defined(self.shape) - def storage_size(self): - raw_size = self.storage_elements() * self.element_size() + def storage_size(self, scale=1.0): + raw_size = self.storage_elements() * self.element_size() * scale if raw_size == 0: raw_size = 1 # force it to take up space rounded_size = numeric_util.round_up(numeric_util.round_up_to_int(raw_size), self.alignment) return rounded_size - def storage_size_for_sub_purpose(self, sub_purpose, param_a=None, param_b=None): + def storage_size_for_sub_purpose(self, arch, sub_purpose, param_a=None, param_b=None): alt_shape = self.storage_shape_for_sub_purpose(sub_purpose, param_a, param_b) elems = shape_num_elements(alt_shape) if elems is None: return 0 if sub_purpose == TensorSubPurpose.DoubleBuffer: - raw_size = elems * self.element_size() * self.compression_scale_for_worst_weight_stream + raw_size = ( + elems + * self.element_size() + * self.compression_scale_for_worst_weight_stream + * arch.weight_estimation_scaling + ) else: # Rolling buffers are used for intermediate data in ifm streaming # These will all use the NHCWB16 format, and need to be aligned to 16 in the C-dimension |