From 568580290c746b81f9808b133be1aa9aeb503856 Mon Sep 17 00:00:00 2001 From: Mike Kelly Date: Mon, 27 Jan 2020 12:14:47 +0000 Subject: IVGCVSW-4380 Add NEON backend support for BatchToSpaceNd * Added NeonBatchToSpaceNdWorkload class. * Added CreateBatchToSpaceNd implementation to NeonWorkloadFactory. * Added IsBatchToSpaceNdSupported implementation to NeonLayerSupport. * Enabled BatchToSpaceNd tests on Neon backend. Signed-off-by: Mike Kelly Change-Id: I179feefaa67dc87a03fcbe52e5df100c1188f9a5 --- src/backends/neon/workloads/CMakeLists.txt | 2 ++ 1 file changed, 2 insertions(+) (limited to 'src/backends/neon/workloads/CMakeLists.txt') diff --git a/src/backends/neon/workloads/CMakeLists.txt b/src/backends/neon/workloads/CMakeLists.txt index 46b5332235..02ffedcf09 100644 --- a/src/backends/neon/workloads/CMakeLists.txt +++ b/src/backends/neon/workloads/CMakeLists.txt @@ -14,6 +14,8 @@ list(APPEND armnnNeonBackendWorkloads_sources NeonArgMinMaxWorkload.hpp NeonBatchNormalizationWorkload.cpp NeonBatchNormalizationWorkload.hpp + NeonBatchToSpaceNdWorkload.cpp + NeonBatchToSpaceNdWorkload.hpp NeonConcatWorkload.cpp NeonConcatWorkload.hpp NeonConstantWorkload.cpp -- cgit v1.2.1