aboutsummaryrefslogtreecommitdiff
path: root/ethosu/vela/mark_tensors.py
diff options
context:
space:
mode:
authorTim Hall <tim.hall@arm.com>2020-06-15 20:47:35 +0100
committerTim Hall <tim.hall@arm.com>2020-06-18 17:53:52 +0100
commitc30f495dc013a73e371dd8053a0381e4707ab309 (patch)
tree902ded0cf05e6d00eb0d1a1f9c1464a1052def92 /ethosu/vela/mark_tensors.py
parent0fc46dc565b6991409169cc40fe9ac74237857fa (diff)
downloadethos-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/mark_tensors.py')
-rw-r--r--ethosu/vela/mark_tensors.py4
1 files changed, 3 insertions, 1 deletions
diff --git a/ethosu/vela/mark_tensors.py b/ethosu/vela/mark_tensors.py
index b557f903..72ab8cfa 100644
--- a/ethosu/vela/mark_tensors.py
+++ b/ethosu/vela/mark_tensors.py
@@ -112,7 +112,9 @@ tensor_purposes = [ # ops, input_purpose
),
(
set(("Conv2DBackpropInputSwitchedBias",)),
- purpose_from_list([TensorPurpose.FeatureMap, TensorPurpose.Weights, TensorPurpose.FeatureMap, TensorPurpose.FeatureMap]),
+ purpose_from_list(
+ [TensorPurpose.FeatureMap, TensorPurpose.Weights, TensorPurpose.FeatureMap, TensorPurpose.FeatureMap]
+ ),
),
(
set(("QuantizedConv2D", "QuantizedMatMul")),