aboutsummaryrefslogtreecommitdiff
path: root/ethosu/vela/high_level_command_stream_generator.py
diff options
context:
space:
mode:
authorJonas Ohlsson <jonas.ohlsson@arm.com>2022-03-30 10:30:25 +0200
committerJonas Ohlsson <jonas.ohlsson@arm.com>2022-03-30 15:54:14 +0200
commitd85750702229af97c0b0bbda6e397a23254b6144 (patch)
tree389962105a35d5cef595cfeb5d640bd59a0d0ff8 /ethosu/vela/high_level_command_stream_generator.py
parentcc5f4de1c35ba44fca7ff6295c6ae846f8242344 (diff)
downloadethos-u-vela-d85750702229af97c0b0bbda6e397a23254b6144.tar.gz
Update version of Black to 22.3.0
Update version of Black to 22.3.0 due to updated dependencies. Updates to fix reported issues due to new version. Signed-off-by: Jonas Ohlsson <jonas.ohlsson@arm.com> Change-Id: I60056aae452093ce8dcea1f499ecced22b25eef1
Diffstat (limited to 'ethosu/vela/high_level_command_stream_generator.py')
-rw-r--r--ethosu/vela/high_level_command_stream_generator.py8
1 files changed, 7 insertions, 1 deletions
diff --git a/ethosu/vela/high_level_command_stream_generator.py b/ethosu/vela/high_level_command_stream_generator.py
index eef4e6d6..81c0d5b4 100644
--- a/ethosu/vela/high_level_command_stream_generator.py
+++ b/ethosu/vela/high_level_command_stream_generator.py
@@ -67,7 +67,13 @@ def generate_high_level_commands_for_sched_op(sched_op, schedule):
ofm_tensor = ps.ofm_tensor
# Get Tensors and Full Shapes
- (ifm_tensor, ifm2_tensor, uncomp_weight_tensor, _, _,) = parent_op.get_ifm_ifm2_weights_biases_ofm()
+ (
+ ifm_tensor,
+ ifm2_tensor,
+ uncomp_weight_tensor,
+ _,
+ _,
+ ) = parent_op.get_ifm_ifm2_weights_biases_ofm()
ifm = sched_op.ifm
ifm2 = sched_op.ifm2
ofm_shape = sched_op.ofm.shape