2 files changed, 6 insertions, 0 deletions
diff --git a/arm_compute/core/CL/kernels/CLMeanStdDevKernel.h b/arm_compute/core/CL/kernels/CLMeanStdDevKernel.h
index 9f30f76e1b..2c09590dab 100644
--- a/arm_compute/core/CL/kernels/CLMeanStdDevKernel.h
+++ b/arm_compute/core/CL/kernels/CLMeanStdDevKernel.h
@@ -63,12 +63,15 @@ public:
     // Inherited methods overridden:
     void run(const Window &window, cl::CommandQueue &queue) override;
 
+    BorderSize border_size() const override;
+
 private:
     const ICLImage *_input;
     float          *_mean;
     float          *_stddev;
     cl::Buffer     *_global_sum;
     cl::Buffer     *_global_sum_squared;
+    BorderSize      _border_size;
 };
 }
 #endif /* __ARM_COMPUTE_CLMEANSTDDEVKERNEL_H__ */
diff --git a/arm_compute/core/NEON/kernels/NEMeanStdDevKernel.h b/arm_compute/core/NEON/kernels/NEMeanStdDevKernel.h
index 83407ccb7d..9c72b20d58 100644
--- a/arm_compute/core/NEON/kernels/NEMeanStdDevKernel.h
+++ b/arm_compute/core/NEON/kernels/NEMeanStdDevKernel.h
@@ -64,6 +64,8 @@ public:
     // Inherited methods overridden:
     void run(const Window &window) override;
 
+    BorderSize border_size() const override;
+
 private:
     const IImage *_input;
     float        *_mean;
@@ -71,6 +73,7 @@ private:
     uint64_t     *_global_sum;
     uint64_t     *_global_sum_squared;
     std::mutex    _mtx;
+    BorderSize    _border_size;
 };
 }
 #endif /* __ARM_COMPUTE_NEMEANSTDDEVKERNEL_H__ */