aboutsummaryrefslogtreecommitdiff
path: root/ethosu/vela/tosa_supported_operators.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/tosa_supported_operators.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/tosa_supported_operators.py')
-rw-r--r--ethosu/vela/tosa_supported_operators.py25
1 files changed, 22 insertions, 3 deletions
diff --git a/ethosu/vela/tosa_supported_operators.py b/ethosu/vela/tosa_supported_operators.py
index e3785113..15e15695 100644
--- a/ethosu/vela/tosa_supported_operators.py
+++ b/ethosu/vela/tosa_supported_operators.py
@@ -38,11 +38,30 @@ class TosaSupportedOperators:
fc_vector_products = set((Op.FullyConnected,))
mac_main_ops = convolution_like_ops | pooling_ops | fc_vector_products
- memory_only_ops = set((Op.Reshape, Op.Transpose, Op.Concat, Op.SplitSliceRead,))
- binary_elem_wise_add_mul_sub = set((Op.Add, Op.Mul, Op.RescaleMul, Op.Sub,))
+ memory_only_ops = set(
+ (
+ Op.Reshape,
+ Op.Transpose,
+ Op.Concat,
+ Op.SplitSliceRead,
+ )
+ )
+ binary_elem_wise_add_mul_sub = set(
+ (
+ Op.Add,
+ Op.Mul,
+ Op.RescaleMul,
+ Op.Sub,
+ )
+ )
elem_wise_ops = binary_elem_wise_add_mul_sub
type_conversion_ops = set((Op.Rescale,))
- relu_ops = set((Op.Clamp, Op.ReluN,))
+ relu_ops = set(
+ (
+ Op.Clamp,
+ Op.ReluN,
+ )
+ )
activation_ops = relu_ops | set((Op.Table,))
pad_ops = set((Op.Pad,))