diff options
Diffstat (limited to 'arm_compute/core')
-rw-r--r-- | arm_compute/core/CL/kernels/CLMeanStdDevKernel.h | 3 | ||||
-rw-r--r-- | arm_compute/core/NEON/kernels/NEMeanStdDevKernel.h | 3 |
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__ */ |