commit 91e4c6d43921b735e6aee78d8611bc0372eda4df
author Nikhil Raj <nikhil.raj@arm.com> Fri Jul 05 12:22:58 2019 +0100
committer Matteo Martincigh <matteo.martincigh@arm.com> Fri Jul 05 12:55:51 2019 +0000
tree 0a1cf9d18ba23984c6a66a79cf8105834b00ef55
parent 328d92b84e6bd29906bf1fc496b1103af7f7e115

IVGCVSW-3293 Add CL backend support for PReLU

Change-Id: I1bb187db89bb3eb883b8f0aca4c3439d82b56583
Signed-off-by: Nikhil Raj <nikhil.raj@arm.com>

src/backends/cl/ClLayerSupport.cpp

diff --git a/src/backends/cl/ClLayerSupport.cpp b/src/backends/cl/ClLayerSupport.cpp
index d79f612..a6baa3c 100644
--- a/src/backends/cl/ClLayerSupport.cpp
+++ b/src/backends/cl/ClLayerSupport.cpp
@@ -39,6 +39,7 @@
 #include "workloads/ClPadWorkload.hpp"
 #include "workloads/ClPermuteWorkload.hpp"
 #include "workloads/ClPooling2dWorkload.hpp"
+#include "workloads/ClPreluWorkload.hpp"
 #include "workloads/ClQuantizeWorkload.hpp"
 #include "workloads/ClSoftmaxBaseWorkload.hpp"
 #include "workloads/ClSpaceToBatchNdWorkload.hpp"
@@ -571,6 +572,14 @@
     FORWARD_WORKLOAD_VALIDATE_FUNC(ClPooling2dWorkloadValidate, reasonIfUnsupported, input, output, descriptor);
 }
 
+bool ClLayerSupport::IsPreluSupported(const armnn::TensorInfo &input,
+                                      const armnn::TensorInfo &alpha,
+                                      const armnn::TensorInfo &output,
+                                      armnn::Optional<std::string &> reasonIfUnsupported) const
+{
+    FORWARD_WORKLOAD_VALIDATE_FUNC(ClPreluWorkloadValidate, reasonIfUnsupported, input, alpha, output);
+}
+
 bool ClLayerSupport::IsQuantizeSupported(const TensorInfo& input,
                                          const TensorInfo& output,
                                          Optional<std::string&> reasonIfUnsupported) const