diff options
author | Fredrik Svedberg <fredrik.svedberg@arm.com> | 2022-08-19 16:06:04 +0200 |
---|---|---|
committer | Fredrik Svedberg <fredrik.svedberg@arm.com> | 2022-08-31 15:32:24 +0200 |
commit | 8ddd4899892dace88306b3b155dbf47cc47fa4cd (patch) | |
tree | 1322377438b283ac8862696f8232ef48cd98a0ea /ethosu/vela/tflite_supported_operators.py | |
parent | 1bbd06bf538ff9d290e9703b6bac05b9617f2ab0 (diff) | |
download | ethos-u-vela-8ddd4899892dace88306b3b155dbf47cc47fa4cd.tar.gz |
MLBEDSW-6832 PReLU support in Vela
Added PReLU support in graph optimiser.
Signed-off-by: Fredrik Svedberg <fredrik.svedberg@arm.com>
Change-Id: I3a188675e3edcdf0b4a4bfcdd134fda0bf8a560f
Diffstat (limited to 'ethosu/vela/tflite_supported_operators.py')
-rw-r--r-- | ethosu/vela/tflite_supported_operators.py | 10 |
1 files changed, 9 insertions, 1 deletions
diff --git a/ethosu/vela/tflite_supported_operators.py b/ethosu/vela/tflite_supported_operators.py index 5d25e37b..1915d43b 100644 --- a/ethosu/vela/tflite_supported_operators.py +++ b/ethosu/vela/tflite_supported_operators.py @@ -123,7 +123,15 @@ class TFLiteSupportedOperators: Op.Clip, ) ) - activation_ops = relu_ops | set((Op.Tanh, Op.Sigmoid, Op.Softmax, Op.HardSwish)) + activation_ops = relu_ops | set( + ( + Op.Tanh, + Op.Sigmoid, + Op.Softmax, + Op.HardSwish, + Op.Prelu, + ) + ) npu_post_ops = ( # activation functions activation_ops |