diff options
Diffstat (limited to 'arm_compute/runtime/NEON/functions/NEGEMMTranspose1xW.h')
-rw-r--r-- | arm_compute/runtime/NEON/functions/NEGEMMTranspose1xW.h | 8 |
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__ */ |