commit 338595bca8ab60492f10626860acb1ab3722b1ce
author Vidhya Sudhan Loganathan <vidhyasudhan.loganathan@arm.com> Fri Jun 28 14:09:53 2019 +0100
committer VidhyaSudhan Loganathan <vidhyasudhan.loganathan@arm.com> Fri Jun 28 14:15:30 2019 +0000
tree 03504ec3a2973e30c80f9bf56b77b4a4c7c9d83c
parent 7026b303d636e7639f8877ae8d5eff54f39c1121

COMPMID-2234 : Add support for axis 3 in NE/CLConcatenateLayer

Change-Id: Ic86f89ece3afe72809bc69c6de6fee7d21daa1d4
Signed-off-by: Vidhya Sudhan Loganathan <vidhyasudhan.loganathan@arm.com>
Reviewed-on: https://review.mlplatform.org/c/1440
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Gian Marco Iodice <gianmarco.iodice@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>

arm_compute/core/NEON/NEKernels.h

diff --git a/arm_compute/core/NEON/NEKernels.h b/arm_compute/core/NEON/NEKernels.h
index 4023d82..e41f299 100644
--- a/arm_compute/core/NEON/NEKernels.h
+++ b/arm_compute/core/NEON/NEKernels.h
@@ -30,6 +30,7 @@
 #include "arm_compute/core/NEON/kernels/NEActivationLayerKernel.h"
 #include "arm_compute/core/NEON/kernels/NEArithmeticAdditionKernel.h"
 #include "arm_compute/core/NEON/kernels/NEArithmeticSubtractionKernel.h"
+#include "arm_compute/core/NEON/kernels/NEBatchConcatenateLayerKernel.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"