From c2449827ec55f49b6087e3e385fb3c4f6776dc6a Mon Sep 17 00:00:00 2001 From: James Peet Date: Mon, 19 Jul 2021 17:09:16 +0100 Subject: MLBEDSW-4892: Fix weight compression of asymmetrically dilated kernels. Signed-off-by: James Peet Change-Id: I5bf39aa4f1fb48bcb0423edc4cd1d01f59aac1db --- ethosu/vela/weight_compressor.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'ethosu/vela') 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, -- cgit v1.2.1