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/npu_performance.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/npu_performance.py')
| -rw-r--r-- | ethosu/vela/npu_performance.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/ethosu/vela/npu_performance.py b/ethosu/vela/npu_performance.py index abe7134d..98626de6 100644 --- a/ethosu/vela/npu_performance.py +++ b/ethosu/vela/npu_performance.py @@ -25,7 +25,6 @@ import numpy as np from . import numeric_util from .architecture_features import Block -from .architecture_features import Kernel from .nn_graph import PassPlacement from .nn_graph import SchedulerRewrite from .operation import NpuBlockType @@ -241,7 +240,9 @@ def performance_metrics_for_pass(arch, ps, block_config=None, rewrite_list=[], f ifm_tensor, _, weight_tensor, ofm_tensor = ps.get_primary_op_ifm_ifm2_weights_ofm() - if npu_block_type in set((NpuBlockType.ConvolutionMxN, NpuBlockType.ConvolutionDepthWise, NpuBlockType.Pooling)): + if npu_block_type in set( + (NpuBlockType.ConvolutionMxN, NpuBlockType.ConvolutionDepthWise, NpuBlockType.Pooling) + ): # extent the ifm to full dimension ifm_tensor_brick_size = tuple(numeric_util.full_shape(4, list(ifm_tensor.brick_size), 1)) ifm_tensor_shape = numeric_util.full_shape(4, ifm_tensor.shape, 1) |