From ce79a9be912fbb98bd66f5dd10c004ec23227f1e Mon Sep 17 00:00:00 2001 From: James Conroy Date: Tue, 12 May 2020 18:08:52 +0100 Subject: IVGCVSW-4451 Add QLstm NEON workload * Adds QLstm workload. * Adds CreateWorkload and Layer tests. Signed-off-by: James Conroy Change-Id: I585eb2691395ee4ccd45b5a853660f90fc5cc821 --- src/backends/neon/NeonWorkloadFactory.hpp | 3 +++ 1 file changed, 3 insertions(+) (limited to 'src/backends/neon/NeonWorkloadFactory.hpp') diff --git a/src/backends/neon/NeonWorkloadFactory.hpp b/src/backends/neon/NeonWorkloadFactory.hpp index d6968fa9dd..0e9428ad24 100644 --- a/src/backends/neon/NeonWorkloadFactory.hpp +++ b/src/backends/neon/NeonWorkloadFactory.hpp @@ -174,6 +174,9 @@ public: std::unique_ptr CreatePrelu(const PreluQueueDescriptor& descriptor, const WorkloadInfo& info) const override; + std::unique_ptr CreateQLstm(const QLstmQueueDescriptor& descriptor, + const WorkloadInfo& info) const override; + std::unique_ptr CreateQuantize(const QuantizeQueueDescriptor& descriptor, const WorkloadInfo& info) const override; -- cgit v1.2.1