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/shared_buffer_allocation.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/shared_buffer_allocation.py')
| -rw-r--r-- | ethosu/vela/shared_buffer_allocation.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/ethosu/vela/shared_buffer_allocation.py b/ethosu/vela/shared_buffer_allocation.py index 72caa1b6..040f2a3b 100644 --- a/ethosu/vela/shared_buffer_allocation.py +++ b/ethosu/vela/shared_buffer_allocation.py @@ -106,7 +106,9 @@ class SharedBufferAllocation: def try_block(self, ofm_block: Block): # Get IFM block configuration ifm_block_depth = ofm_block.depth if self.is_equal_depth_op else self.ifm_block_depth - ifm_block = self.arch.get_ifm_block_size(ifm_block_depth, ofm_block, self.kernel, ifm_resampling_mode=self.ifm_resampling_mode) + ifm_block = self.arch.get_ifm_block_size( + ifm_block_depth, ofm_block, self.kernel, ifm_resampling_mode=self.ifm_resampling_mode + ) ifm_config = self.arch.get_block_config(ifm_block.width, ifm_block.height, ifm_block.depth) if ifm_config is None: return None |