diff options
| author | James Peet <james.peet@arm.com> | 2021-07-19 17:09:16 +0100 |
|---|---|---|
| committer | James Peet <james.peet@arm.com> | 2021-07-26 12:05:57 +0100 |
| commit | c2449827ec55f49b6087e3e385fb3c4f6776dc6a (patch) | |
| tree | d8de5a99c17b6e418028429eadc9da3cfb286345 | |
| parent | 7519d50c993d60faa1ea09e56abfbf17cef23b49 (diff) | |
| download | ethos-u-vela-c2449827ec55f49b6087e3e385fb3c4f6776dc6a.tar.gz | |
MLBEDSW-4892: Fix weight compression of asymmetrically dilated kernels.
Signed-off-by: James Peet <james.peet@arm.com>
Change-Id: I5bf39aa4f1fb48bcb0423edc4cd1d01f59aac1db
| -rw-r--r-- | ethosu/vela/weight_compressor.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/ethosu/vela/weight_compressor.py b/ethosu/vela/weight_compressor.py index 65361436..10a1a6dd 100644 --- a/ethosu/vela/weight_compressor.py +++ b/ethosu/vela/weight_compressor.py @@ -149,8 +149,8 @@ def encode_weights( ifm_ublock = ArchitectureFeatures.accelerator_configs[accelerator].ifm_ublock ofm_ublock = ArchitectureFeatures.accelerator_configs[accelerator].ofm_ublock - decomp_h = ArchitectureFeatures.SubKernelMax.height // dilation_xy[0] - decomp_w = ArchitectureFeatures.SubKernelMax.width // dilation_xy[1] + decomp_h = ArchitectureFeatures.SubKernelMax.height // dilation_xy[1] + decomp_w = ArchitectureFeatures.SubKernelMax.width // dilation_xy[0] return mlw_codec.reorder_encode( ifm_ublock.depth, |