aboutsummaryrefslogtreecommitdiff
path: root/ethosu/vela/softmax.py
diff options
context:
space:
mode:
authorJonas Ohlsson <jonas.ohlsson@arm.com>2022-03-04 14:58:56 +0100
committerJonas Ohlsson <jonas.ohlsson@arm.com>2022-03-04 15:05:22 +0100
commit25e700c9894bc9ca5fb219e902c907c04b2048b2 (patch)
treefc90f128814f5c642e54a4c9c49af6c63c226ff2 /ethosu/vela/softmax.py
parent51bb3379a24139e2af234548e56ff063728f8e7e (diff)
downloadethos-u-vela-25e700c9894bc9ca5fb219e902c907c04b2048b2.tar.gz
MLBEDSW-3367 Update pre-commit flake8 version
Update the version of flake8 used in pre-commit to facilitate adding mypy to pre-commit. Signed-off-by: Jonas Ohlsson <jonas.ohlsson@arm.com> Change-Id: I457dec87b77487ca6f14ff4a679c4cc927b272b0
Diffstat (limited to 'ethosu/vela/softmax.py')
-rw-r--r--ethosu/vela/softmax.py4
1 files changed, 1 insertions, 3 deletions
diff --git a/ethosu/vela/softmax.py b/ethosu/vela/softmax.py
index 13ca319b..711c1e04 100644
--- a/ethosu/vela/softmax.py
+++ b/ethosu/vela/softmax.py
@@ -501,9 +501,7 @@ class SoftMax:
reciprocal_right_shift = add_op_get_ofm(create_sub(name, const_31, headroom_plus_one, no_scale_quant))
# PASS 7 - SHL
- one = create_const_tensor(
- f"one_const", [1, 1, 1, 1], DataType.int32, [1], np.int32, quantization=no_scale_quant
- )
+ one = create_const_tensor("one_const", [1, 1, 1, 1], DataType.int32, [1], np.int32, quantization=no_scale_quant)
constant_one = add_op_get_ofm(
create_shl(f"{self.op.name}_shl{pass_number}", one, reciprocal_right_shift, no_scale_quant)
)