aboutsummaryrefslogtreecommitdiff
path: root/arm_compute/runtime/NEON/functions/NEDepthConvertLayer.h
diff options
context:
space:
mode:
Diffstat (limited to 'arm_compute/runtime/NEON/functions/NEDepthConvertLayer.h')
-rw-r--r--arm_compute/runtime/NEON/functions/NEDepthConvertLayer.h6
1 files changed, 3 insertions, 3 deletions
diff --git a/arm_compute/runtime/NEON/functions/NEDepthConvertLayer.h b/arm_compute/runtime/NEON/functions/NEDepthConvertLayer.h
index 1fdad30115..ebb9530c71 100644
--- a/arm_compute/runtime/NEON/functions/NEDepthConvertLayer.h
+++ b/arm_compute/runtime/NEON/functions/NEDepthConvertLayer.h
@@ -25,7 +25,7 @@
#define __ARM_COMPUTE_NEDEPTHCONVERT_H__
#include "arm_compute/core/Types.h"
-#include "arm_compute/runtime/NEON/INESimpleFunction.h"
+#include "arm_compute/runtime/NEON/INESimpleFunctionNoBorder.h"
#include <cstdint>
@@ -34,7 +34,7 @@ namespace arm_compute
class ITensor;
/**Basic function to run @ref NEDepthConvertLayerKernel */
-class NEDepthConvertLayer : public INESimpleFunction
+class NEDepthConvertLayer : public INESimpleFunctionNoBorder
{
public:
/* Contructor */
@@ -69,5 +69,5 @@ public:
*/
static Status validate(const ITensorInfo *input, const ITensorInfo *output, ConvertPolicy policy, uint32_t shift = 0);
};
-}
+} // namespace arm_compute
#endif /*__ARM_COMPUTE_NEDEPTHCONVERT_H__*/