aboutsummaryrefslogtreecommitdiff
path: root/arm_compute/core/NEON/NEKernels.h
diff options
context:
space:
mode:
authorgiuros01 <giuseppe.rossini@arm.com>2019-02-19 13:53:10 +0000
committerGiuseppe Rossini <giuseppe.rossini@arm.com>2019-03-01 10:25:15 +0000
commitba3682521398f228d6db6fb1bfbe464536b8d002 (patch)
tree07ed22a26977d94f1df6699ab940519ba5999942 /arm_compute/core/NEON/NEKernels.h
parent6e6179443a1e8413dd209e756b8393a5e54243e1 (diff)
downloadComputeLibrary-ba3682521398f228d6db6fb1bfbe464536b8d002.tar.gz
COMPMID-1947: Implement NESpaceToBatch
Change-Id: I59b3c17874ba24559b7fddf74f7659a1b9177759 Signed-off-by: giuros01 <giuseppe.rossini@arm.com> Reviewed-on: https://review.mlplatform.org/c/735 Tested-by: Arm Jenkins <bsgcomp@arm.com> Reviewed-by: Michalis Spyrou <michalis.spyrou@arm.com>
Diffstat (limited to 'arm_compute/core/NEON/NEKernels.h')
-rw-r--r--arm_compute/core/NEON/NEKernels.h1
1 files changed, 1 insertions, 0 deletions
diff --git a/arm_compute/core/NEON/NEKernels.h b/arm_compute/core/NEON/NEKernels.h
index cc6af39a54..6509e39d7d 100644
--- a/arm_compute/core/NEON/NEKernels.h
+++ b/arm_compute/core/NEON/NEKernels.h
@@ -120,6 +120,7 @@
#include "arm_compute/core/NEON/kernels/NESobel5x5Kernel.h"
#include "arm_compute/core/NEON/kernels/NESobel7x7Kernel.h"
#include "arm_compute/core/NEON/kernels/NESoftmaxLayerKernel.h"
+#include "arm_compute/core/NEON/kernels/NESpaceToBatchLayerKernel.h"
#include "arm_compute/core/NEON/kernels/NEStackLayerKernel.h"
#include "arm_compute/core/NEON/kernels/NEStridedSliceKernel.h"
#include "arm_compute/core/NEON/kernels/NETableLookupKernel.h"