aboutsummaryrefslogtreecommitdiff
path: root/ethosu/vela/tosa_reader.py
diff options
context:
space:
mode:
authorPer Åstrand <per.astrand@arm.com>2024-04-22 11:48:09 +0200
committerJohan Alfven <johan.alfven@arm.com>2024-04-23 18:56:57 +0200
commitbab7f28a65a0bf940b0cc61be48bbe478dde51f7 (patch)
treec998d80b8a040f3039a53e07d473c40b2bfc73bb /ethosu/vela/tosa_reader.py
parent92240e7979018a197b42aab2da16dc002d86f224 (diff)
downloadethos-u-vela-bab7f28a65a0bf940b0cc61be48bbe478dde51f7.tar.gz
Reformat code to align with precommit
Signed-off-by: Per Åstrand <per.astrand@arm.com> Change-Id: Idc6f6959bfc7eabce2f5b6e0d4935d292dcf6618
Diffstat (limited to 'ethosu/vela/tosa_reader.py')
-rw-r--r--ethosu/vela/tosa_reader.py3
1 files changed, 2 insertions, 1 deletions
diff --git a/ethosu/vela/tosa_reader.py b/ethosu/vela/tosa_reader.py
index 5ffefade..670b264a 100644
--- a/ethosu/vela/tosa_reader.py
+++ b/ethosu/vela/tosa_reader.py
@@ -247,7 +247,8 @@ class TosaSubgraph:
if depth_multiplier > 1:
assert op.ifm.shape[-1] == 1 and op.ofm.shape[-1] == depth_multiplier, (
"For depth multipliers > 1, IFM channels must be 1 and "
- "OFM channels must be equal to the depth multiplier")
+ "OFM channels must be equal to the depth multiplier"
+ )
op.attrs["depth_multiplier"] = depth_multiplier
if op.type == Op.SplitSliceRead:
op.read_offsets[0] = Shape4D.from_list(list(op.attrs["start"]), 0)