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/architecture_features.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/architecture_features.py')
| -rw-r--r-- | ethosu/vela/architecture_features.py | 11 |
1 files changed, 8 insertions, 3 deletions
diff --git a/ethosu/vela/architecture_features.py b/ethosu/vela/architecture_features.py index 15cf0ea4..58ab0ad3 100644 --- a/ethosu/vela/architecture_features.py +++ b/ethosu/vela/architecture_features.py @@ -342,9 +342,14 @@ Note the difference between ArchitectureFeatures and CompilerOptions return min(max_block_depth, ifm_depth) # Calculate the size of the IFM block given a depth, target OFM block and a kernel - def get_ifm_block_size(self, ifm_block_depth, ofm_block: Block, - kernel: Kernel, subkernel: Block = Block(8, 8, 65536), - ifm_resampling_mode=resampling_mode.NONE): + def get_ifm_block_size( + self, + ifm_block_depth, + ofm_block: Block, + kernel: Kernel, + subkernel: Block = Block(8, 8, 65536), + ifm_resampling_mode=resampling_mode.NONE, + ): upscaling = 1 if ifm_resampling_mode == resampling_mode.NONE else 2 # Height ifm_odd_2x_height_enable = 0 |