diff options
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 |