aboutsummaryrefslogtreecommitdiff
path: root/src/backends/neon/NeonWorkloadFactory.hpp
diff options
context:
space:
mode:
authorMatthew Jackson <matthew.jackson@arm.com>2019-08-01 10:01:34 +0100
committerMatthew Jackson <matthew.jackson@arm.com>2019-08-02 14:16:54 +0100
commit87f65eab4abb65273ea11eb8ca876196ef82c6c5 (patch)
tree227e8c04d4531aed64b49f20423b53465fb33526 /src/backends/neon/NeonWorkloadFactory.hpp
parent9c3cae8683e4b24932446b88d3ecbc02f9f9fa08 (diff)
downloadarmnn-87f65eab4abb65273ea11eb8ca876196ef82c6c5.tar.gz
IVGCVSW-3539 Add Neon support and tests for Stack
* Added Neon backend support for Stack * Added unit tests for Stack on the Neon backend Signed-off-by: Matthew Jackson <matthew.jackson@arm.com> Change-Id: I21379966a6303285f8b65418d0d4ac7982cc9f04
Diffstat (limited to 'src/backends/neon/NeonWorkloadFactory.hpp')
-rw-r--r--src/backends/neon/NeonWorkloadFactory.hpp3
1 files changed, 3 insertions, 0 deletions
diff --git a/src/backends/neon/NeonWorkloadFactory.hpp b/src/backends/neon/NeonWorkloadFactory.hpp
index 4fd9bf22ea..d6a1b74941 100644
--- a/src/backends/neon/NeonWorkloadFactory.hpp
+++ b/src/backends/neon/NeonWorkloadFactory.hpp
@@ -180,6 +180,9 @@ public:
std::unique_ptr<IWorkload> CreateGather(const GatherQueueDescriptor& descriptor,
const WorkloadInfo& info) const override;
+ std::unique_ptr<IWorkload> CreateStack(const StackQueueDescriptor& descriptor,
+ const WorkloadInfo& info) const override;
+
private:
mutable std::shared_ptr<NeonMemoryManager> m_MemoryManager;
};