IVGCVSW-6061 Add PRELU support to TF Lite delegate
* Alpha is supported both as a constant and as an
input tensor.
Signed-off-by: James Conroy <james.conroy@arm.com>
Change-Id: I20f30d479b87dc5fbcf75a1ce5305d70ae9b0646
diff --git a/delegate/src/armnn_delegate.cpp b/delegate/src/armnn_delegate.cpp
index 7e4f5b5..0c984ec 100644
--- a/delegate/src/armnn_delegate.cpp
+++ b/delegate/src/armnn_delegate.cpp
@@ -24,6 +24,7 @@
#include "Pack.hpp"
#include "Pad.hpp"
#include "Pooling.hpp"
+#include "Prelu.hpp"
#include "Quantization.hpp"
#include "Redefine.hpp"
#include "Reduce.hpp"
@@ -733,11 +734,11 @@
nodeIndex,
kTfLiteBuiltinPadv2);
case kTfLiteBuiltinPrelu:
- return VisitActivationOperator(delegateData,
- tfLiteContext,
- tfLiteNode,
- nodeIndex,
- kTfLiteBuiltinPrelu);
+ return VisitPreluOperator(delegateData,
+ tfLiteContext,
+ tfLiteNode,
+ nodeIndex,
+ kTfLiteBuiltinPrelu);
case kTfLiteBuiltinQuantize:
return VisitQuantizeOperator(delegateData,
tfLiteContext,