commit 2f00b7421f259851448a0be72e9eb01141faad2e
author Aron Virginas-Tar <Aron.Virginas-Tar@arm.com> Mon Sep 30 13:28:08 2019 +0100
committer Kevin May <kevin.may@arm.com> Mon Sep 30 14:13:42 2019 +0000
tree 109902951005eca1b6bac148c7b41387e487432d
parent 7be47efac07b6276f02a17cb486f9061a426a837

IVGCVSW-3734 Add NEON workload for DepthToSpace

Signed-off-by: Aron Virginas-Tar <Aron.Virginas-Tar@arm.com>
Change-Id: I4fa8cba57f3a2277112c02062f4d2790089d1eb5

src/backends/neon/NeonWorkloadFactory.cpp

diff --git a/src/backends/neon/NeonWorkloadFactory.cpp b/src/backends/neon/NeonWorkloadFactory.cpp
index fb81008..f0b738c 100644
--- a/src/backends/neon/NeonWorkloadFactory.cpp
+++ b/src/backends/neon/NeonWorkloadFactory.cpp
@@ -170,6 +170,12 @@
                                                        m_MemoryManager->GetIntraLayerManager());
 }
 
+std::unique_ptr<IWorkload> NeonWorkloadFactory::CreateDepthToSpace(const DepthToSpaceQueueDescriptor& descriptor,
+                                                                   const WorkloadInfo& info) const
+{
+    return std::make_unique<NeonDepthToSpaceWorkload>(descriptor, info);
+}
+
 std::unique_ptr<IWorkload> NeonWorkloadFactory::CreateDepthwiseConvolution2d(
     const DepthwiseConvolution2dQueueDescriptor& descriptor, const WorkloadInfo& info) const
 {