aboutsummaryrefslogtreecommitdiff
path: root/src/backends/neon/NeonWorkloadFactory.hpp
diff options
context:
space:
mode:
authorEllen Norris-Thompson <ellen.norris-thompson@arm.com>2019-06-26 16:40:36 +0100
committerJames Conroy <james.conroy@arm.com>2019-07-09 13:29:58 +0000
commit2979457c5974e298033c387a5cd33f95e3301623 (patch)
treedf7b6b54def6c608602f487e4cffa0b1dc187a3f /src/backends/neon/NeonWorkloadFactory.hpp
parent39f4639a79625c8f37c6ca547dadc7925378ee3e (diff)
downloadarmnn-2979457c5974e298033c387a5cd33f95e3301623.tar.gz
IVGCVSW-3355 Add NeonSpaceToDepthWorkload
* Add Neon backend support for SpaceToDepth * Enabled Neon Unit, CreateWorkload and EndToEnd tests for SpaceToDepth. * Added QSymm16 support to NeonTensorHandle. Signed-off-by: Ellen Norris-Thompson <ellen.norris-thompson@arm.com> Change-Id: Ibbb0bce5ad77dace2bf4c4c111006d2fadf844e8
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 234167324f..4fd9bf22ea 100644
--- a/src/backends/neon/NeonWorkloadFactory.hpp
+++ b/src/backends/neon/NeonWorkloadFactory.hpp
@@ -123,6 +123,9 @@ public:
std::unique_ptr<IWorkload> CreateSpaceToBatchNd(const SpaceToBatchNdQueueDescriptor& descriptor,
const WorkloadInfo& info) const override;
+ std::unique_ptr<IWorkload> CreateSpaceToDepth(const SpaceToDepthQueueDescriptor& descriptor,
+ const WorkloadInfo& info) const override;
+
std::unique_ptr<IWorkload> CreateFloor(const FloorQueueDescriptor& descriptor,
const WorkloadInfo& info) const override;