From 0be3a88518a73819aaf71ef71f4fb2fbfc14bc88 Mon Sep 17 00:00:00 2001 From: Mike Kelly Date: Fri, 24 Jan 2020 11:27:50 +0000 Subject: IVGCVSW-4374 Add NEON backend support for SpaceToBatchNd * Added NeonSpaceToBatchNdWorkload class. * Added CreateSpaceToBatchNd implementation to NeonWorkloadFactory. * Added IsSpaceToBatchNdSupported implementation to NeonLayerSupport. * Enabled SpaceToBatchNd tests on Neon backend. Signed-off-by: Mike Kelly Change-Id: Ic7d469a380482ec9febd44450f72dd7e53cda03f --- src/backends/neon/backend.mk | 1 + 1 file changed, 1 insertion(+) (limited to 'src/backends/neon/backend.mk') diff --git a/src/backends/neon/backend.mk b/src/backends/neon/backend.mk index 3ddc79ada3..05dcd02c9a 100644 --- a/src/backends/neon/backend.mk +++ b/src/backends/neon/backend.mk @@ -59,6 +59,7 @@ BACKEND_SOURCES := \ workloads/NeonSoftmaxBaseWorkload.cpp \ workloads/NeonSoftmaxFloatWorkload.cpp \ workloads/NeonSoftmaxUint8Workload.cpp \ + workloads/NeonSpaceToBatchNdWorkload.cpp \ workloads/NeonSpaceToDepthWorkload.cpp \ workloads/NeonSplitterWorkload.cpp \ workloads/NeonStackWorkload.cpp \ -- cgit v1.2.1