commit 95abfddfa08ab85d4f88c6f4d2e077969178f2d5
author Michalis Spyrou <michalis.spyrou@arm.com> Wed Nov 28 14:59:47 2018 +0000
committer Michalis Spyrou <michalis.spyrou@arm.com> Fri Nov 30 15:15:07 2018 +0000
tree dd6124e42635423cf01b4b9face5c983c67cb0e4
parent eb6aad7e1eee0c0659ea71444ed4093c52267af3

COMPMID-1815 Remove templates from NEFillborderKernel and create INESimpleFunctionNoBorder

Change-Id: Ia9fdc75b23e9a6208058f8406fb7b5fcd917de2c
Reviewed-on: https://review.mlplatform.org/311
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com>

arm_compute/runtime/NEON/functions/NEFlattenLayer.h

diff --git a/arm_compute/runtime/NEON/functions/NEFlattenLayer.h b/arm_compute/runtime/NEON/functions/NEFlattenLayer.h
index 26d7c7f..3365b35 100644
--- a/arm_compute/runtime/NEON/functions/NEFlattenLayer.h
+++ b/arm_compute/runtime/NEON/functions/NEFlattenLayer.h
@@ -25,14 +25,14 @@
 #define __ARM_COMPUTE_NEFLATTENLAYER_H__
 
 #include "arm_compute/core/Types.h"
-#include "arm_compute/runtime/NEON/INESimpleFunction.h"
+#include "arm_compute/runtime/NEON/INESimpleFunctionNoBorder.h"
 
 namespace arm_compute
 {
 class ITensor;
 
 /** Basic function to execute flatten layer kernel. */
-class NEFlattenLayer : public INESimpleFunction
+class NEFlattenLayer : public INESimpleFunctionNoBorder
 {
 public:
     /** Initialise the kernel's input and output.