commit cc34093a57ea486e48f9aa66fc8b98a7bbdefef1
author James Conroy <james.conroy@arm.com> Tue May 12 18:08:52 2020 +0100
committer James Conroy <james.conroy@arm.com> Wed May 20 13:09:45 2020 +0100
tree 9c4d792e8c52b20fe8541108377d2b314c2232bd
parent 36eed46563a2a254e674865eb8ede62a1558c997

IVGCVSW-4451 Add QLstm NEON workload

* Adds QLstm workload.
* Adds CreateWorkload and Layer tests.

Signed-off-by: James Conroy <james.conroy@arm.com>
Change-Id: I585eb2691395ee4ccd45b5a853660f90fc5cc821

src/backends/neon/NeonWorkloadFactory.hpp

diff --git a/src/backends/neon/NeonWorkloadFactory.hpp b/src/backends/neon/NeonWorkloadFactory.hpp
index d6968fa..0e9428a 100644
--- a/src/backends/neon/NeonWorkloadFactory.hpp
+++ b/src/backends/neon/NeonWorkloadFactory.hpp
@@ -174,6 +174,9 @@
     std::unique_ptr<IWorkload> CreatePrelu(const PreluQueueDescriptor& descriptor,
                                            const WorkloadInfo& info) const override;
 
+    std::unique_ptr<IWorkload> CreateQLstm(const QLstmQueueDescriptor& descriptor,
+                                           const WorkloadInfo& info) const override;
+
     std::unique_ptr<IWorkload> CreateQuantize(const QuantizeQueueDescriptor& descriptor,
                                               const WorkloadInfo& info) const override;