commit ab9e52563f624d9782b97400f643d2632cc8d770
author Matteo Martincigh <matteo.martincigh@arm.com> Thu Jun 13 17:27:46 2019 +0100
committer Matteo Martincigh <matteo.martincigh@arm.com> Wed Jun 19 11:12:35 2019 +0000
tree 5fec7f9e398dbd2337241cd3cc908f7b0e1d588b
parent bee4bc944aa50782ff22cb4a31fbc611212a5e89

IVGCVSW-3268 Add Reference workload support for the new Prelu Activation layer

 * Added reference workload for the PReLU Activation layer
 * Added factory methods
 * Added validation support
 * Added Int16 support
 * Added unit tests

Change-Id: Ic950d908c5e0a335dccd2960a3ffab0f8b599876
Signed-off-by: Matteo Martincigh <matteo.martincigh@arm.com>

src/backends/reference/RefWorkloadFactory.hpp

diff --git a/src/backends/reference/RefWorkloadFactory.hpp b/src/backends/reference/RefWorkloadFactory.hpp
index 78f6bab..333a9ca 100644
--- a/src/backends/reference/RefWorkloadFactory.hpp
+++ b/src/backends/reference/RefWorkloadFactory.hpp
@@ -187,6 +187,9 @@
     std::unique_ptr<IWorkload> CreateQuantize(const QuantizeQueueDescriptor& descriptor,
                                               const WorkloadInfo& info) const override;
 
+    std::unique_ptr<IWorkload> CreatePrelu(const PreluQueueDescriptor& descriptor,
+                                           const WorkloadInfo& info) const override;
+
 private:
 
     template <typename F32Workload, typename U8Workload, typename QueueDescriptorType>