From fc1da1391679c51209c611e95d60569ce4da15cb Mon Sep 17 00:00:00 2001 From: giuros01 Date: Mon, 18 Feb 2019 16:48:35 +0000 Subject: COMPMID-1946: Implement NEBatchToSpace Change-Id: I119645eb3ea437c7dfe59545da58b328a7184f3f Signed-off-by: giuros01 Reviewed-on: https://review.mlplatform.org/c/734 Tested-by: Arm Jenkins Reviewed-by: Georgios Pinitas --- arm_compute/core/NEON/NEKernels.h | 1 + 1 file changed, 1 insertion(+) (limited to 'arm_compute/core/NEON/NEKernels.h') diff --git a/arm_compute/core/NEON/NEKernels.h b/arm_compute/core/NEON/NEKernels.h index 6509e39d7d..8b37b2f603 100644 --- a/arm_compute/core/NEON/NEKernels.h +++ b/arm_compute/core/NEON/NEKernels.h @@ -31,6 +31,7 @@ #include "arm_compute/core/NEON/kernels/NEArithmeticAdditionKernel.h" #include "arm_compute/core/NEON/kernels/NEArithmeticSubtractionKernel.h" #include "arm_compute/core/NEON/kernels/NEBatchNormalizationLayerKernel.h" +#include "arm_compute/core/NEON/kernels/NEBatchToSpaceLayerKernel.h" #include "arm_compute/core/NEON/kernels/NEBitwiseAndKernel.h" #include "arm_compute/core/NEON/kernels/NEBitwiseNotKernel.h" #include "arm_compute/core/NEON/kernels/NEBitwiseOrKernel.h" -- cgit v1.2.1