From f5c07c45b48717fb6391adb35cb73ca7cd5734c3 Mon Sep 17 00:00:00 2001 From: Fredrik Svedberg Date: Fri, 23 Apr 2021 14:36:42 +0200 Subject: [MLBEDSW-4254] Improve weight information in summary Improved weight information showed in summary if --verbose-weights option is used. Signed-off-by: Fredrik Svedberg Change-Id: Iac142f2a813bf1c05aa9da3f8a384466e2914d06 --- ethosu/vela/api.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) (limited to 'ethosu/vela/api.py') diff --git a/ethosu/vela/api.py b/ethosu/vela/api.py index f972133d..e91c0bdb 100644 --- a/ethosu/vela/api.py +++ b/ethosu/vela/api.py @@ -416,15 +416,16 @@ def npu_encode_weights( :param ofm_block_depth: the depth of blocks for processing :param is_depthwise: a boolean indicating these weights are used for a depthwise traversal :param block_traversal: indicates how these weights are traversed on sub-kernel basis - :return: a bytearray of compressed weights + :return: a bytearray of encoded weights """ from .architecture_features import Accelerator from . import weight_compressor acc = Accelerator.from_npu_accelerator(accelerator) - return weight_compressor.encode_weights( + encoded_weights, _ = weight_compressor.encode_weights( acc, weights_volume, dilation_xy, ifm_bitdepth, ofm_block_depth, is_depthwise, block_traversal ) + return encoded_weights def npu_encode_bias(bias: numpy.int64, scale: int, shift: int): -- cgit v1.2.1