1 files changed, 5 insertions, 3 deletions
diff --git a/arm_compute/runtime/NEON/functions/NEGEMMTranspose1xW.h b/arm_compute/runtime/NEON/functions/NEGEMMTranspose1xW.h
index 3f8e731d01..b44c5a3ee3 100644
--- a/arm_compute/runtime/NEON/functions/NEGEMMTranspose1xW.h
+++ b/arm_compute/runtime/NEON/functions/NEGEMMTranspose1xW.h
@@ -24,16 +24,18 @@
 #ifndef __ARM_COMPUTE_NEGEMMTRANSPOSE1XW_H__
 #define __ARM_COMPUTE_NEGEMMTRANSPOSE1XW_H__
 
-#include "arm_compute/runtime/NEON/INESimpleFunction.h"
+#include "arm_compute/runtime/NEON/INESimpleFunctionNoBorder.h"
 
 namespace arm_compute
 {
+class ITensor;
+
 /** Basic function to execute NEGEMMTranspose1xWKernel. This function calls the following NEON kernels:
  *
  *  -# @ref NEGEMMTranspose1xWKernel
  *
  */
-class NEGEMMTranspose1xW : public INESimpleFunction
+class NEGEMMTranspose1xW : public INESimpleFunctionNoBorder
 {
 public:
     /** Initialise the kernel's inputs, output
@@ -51,5 +53,5 @@ public:
      */
     static Status validate(const ITensorInfo *input, const ITensorInfo *output);
 };
-}
+} // namespace arm_compute
 #endif /*__ARM_COMPUTE_NEGEMMTRANSPOSE1XW_H__ */