diff options
author | Jonas Ohlsson <jonas.ohlsson@arm.com> | 2022-03-30 10:30:25 +0200 |
---|---|---|
committer | Jonas Ohlsson <jonas.ohlsson@arm.com> | 2022-03-30 15:54:14 +0200 |
commit | d85750702229af97c0b0bbda6e397a23254b6144 (patch) | |
tree | 389962105a35d5cef595cfeb5d640bd59a0d0ff8 /ethosu/vela/tosa_supported_operators.py | |
parent | cc5f4de1c35ba44fca7ff6295c6ae846f8242344 (diff) | |
download | ethos-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.py | 25 |
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,)) |