aboutsummaryrefslogtreecommitdiff
path: root/arm_compute/runtime/NEON/functions/NEYOLOLayer.h
diff options
context:
space:
mode:
Diffstat (limited to 'arm_compute/runtime/NEON/functions/NEYOLOLayer.h')
-rw-r--r--arm_compute/runtime/NEON/functions/NEYOLOLayer.h4
1 files changed, 2 insertions, 2 deletions
diff --git a/arm_compute/runtime/NEON/functions/NEYOLOLayer.h b/arm_compute/runtime/NEON/functions/NEYOLOLayer.h
index e09dd42f8f..0adc0f1d9a 100644
--- a/arm_compute/runtime/NEON/functions/NEYOLOLayer.h
+++ b/arm_compute/runtime/NEON/functions/NEYOLOLayer.h
@@ -24,7 +24,7 @@
#ifndef __ARM_COMPUTE_NEYOLOLAYER_H__
#define __ARM_COMPUTE_NEYOLOLAYER_H__
-#include "arm_compute/runtime/NEON/INESimpleFunction.h"
+#include "arm_compute/runtime/NEON/INESimpleFunctionNoBorder.h"
#include "arm_compute/core/NEON/kernels/NEYOLOLayerKernel.h"
#include "arm_compute/core/Types.h"
@@ -34,7 +34,7 @@ namespace arm_compute
class ITensor;
/** Basic function to run @ref NEYOLOLayerKernel */
-class NEYOLOLayer : public INESimpleFunction
+class NEYOLOLayer : public INESimpleFunctionNoBorder
{
public:
/** Set the input and output tensor.