diff options
| author | Tim Hall <tim.hall@arm.com> | 2020-06-15 20:47:35 +0100 |
|---|---|---|
| committer | Tim Hall <tim.hall@arm.com> | 2020-06-18 17:53:52 +0100 |
| commit | c30f495dc013a73e371dd8053a0381e4707ab309 (patch) | |
| tree | 902ded0cf05e6d00eb0d1a1f9c1464a1052def92 /ethosu/vela/weight_compressor.py | |
| parent | 0fc46dc565b6991409169cc40fe9ac74237857fa (diff) | |
| download | ethos-u-vela-c30f495dc013a73e371dd8053a0381e4707ab309.tar.gz | |
Code clean-up using black and flake8
- No functional change
Signed-off-by: Tim Hall <tim.hall@arm.com>
Change-Id: I5ab1198b9d092cd041fa9b85b2dee9900d299bfc
Diffstat (limited to 'ethosu/vela/weight_compressor.py')
| -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 df2b0573..c5f4ce10 100644 --- a/ethosu/vela/weight_compressor.py +++ b/ethosu/vela/weight_compressor.py @@ -19,6 +19,7 @@ import math from collections import namedtuple import numpy as np +from ethosu import mlw_codec from .data_type import DataType from .errors import UnsupportedFeatureError @@ -31,7 +32,6 @@ from .tensor import TensorBlockTraversal from .tensor import TensorFormat from .tensor import TensorPurpose from .tensor import TensorSubPurpose -from ethosu import mlw_codec # Contains meta info for a weight compression. If two tensors have identical weight compression config, @@ -236,7 +236,7 @@ def compress_weights(arch, nng, tens, npu_block_type, ofm_block_depth, ofm_depth if tens.consumer_list[0].type == "Conv2DBackpropInputSwitchedBias": # Transpose Convoluion, reverse weights in H and W axes - weights = np.flip(weights, axis=(0,1)) + weights = np.flip(weights, axis=(0, 1)) # Slice weight stream up depth-ways into bricks and compress full_ofm_depth = quant_buf.shape[-1] |