commit 8ddd4899892dace88306b3b155dbf47cc47fa4cd
author Fredrik Svedberg <fredrik.svedberg@arm.com> Fri Aug 19 16:06:04 2022 +0200
committer Fredrik Svedberg <fredrik.svedberg@arm.com> Wed Aug 31 15:32:24 2022 +0200
tree 1322377438b283ac8862696f8232ef48cd98a0ea
parent 1bbd06bf538ff9d290e9703b6bac05b9617f2ab0

MLBEDSW-6832 PReLU support in Vela

Added PReLU support in graph optimiser.

Signed-off-by: Fredrik Svedberg <fredrik.svedberg@arm.com>
Change-Id: I3a188675e3edcdf0b4a4bfcdd134fda0bf8a560f

ethosu/vela/tflite_supported_operators.py

diff --git a/ethosu/vela/tflite_supported_operators.py b/ethosu/vela/tflite_supported_operators.py
index 5d25e37..1915d43 100644
--- a/ethosu/vela/tflite_supported_operators.py
+++ b/ethosu/vela/tflite_supported_operators.py
@@ -123,7 +123,15 @@
             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