aboutsummaryrefslogtreecommitdiff
path: root/ethosu/vela/tflite_supported_operators.py
diff options
context:
space:
mode:
Diffstat (limited to 'ethosu/vela/tflite_supported_operators.py')
-rw-r--r--ethosu/vela/tflite_supported_operators.py10
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