From a2611815f278334c801094d095901d36e111c3f9 Mon Sep 17 00:00:00 2001 From: Giorgio Arena Date: Fri, 21 Jul 2017 10:08:48 +0100 Subject: COMPMID-417 NEON/CL MeanStdDev bugfix using FillBorderKernel Change-Id: Ic48ba7f69783d0e1e80611264e2bc67d1732436e Reviewed-on: http://mpd-gerrit.cambridge.arm.com/81293 Reviewed-by: Anthony Barbier Tested-by: Kaizen --- src/core/NEON/kernels/NEMeanStdDevKernel.cpp | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) (limited to 'src/core/NEON/kernels/NEMeanStdDevKernel.cpp') diff --git a/src/core/NEON/kernels/NEMeanStdDevKernel.cpp b/src/core/NEON/kernels/NEMeanStdDevKernel.cpp index 4616203d66..1eb7e45e36 100644 --- a/src/core/NEON/kernels/NEMeanStdDevKernel.cpp +++ b/src/core/NEON/kernels/NEMeanStdDevKernel.cpp @@ -85,10 +85,15 @@ std::pair accumulate(const Window &window, Iterator &ite } // namespace NEMeanStdDevKernel::NEMeanStdDevKernel() - : _input(nullptr), _mean(nullptr), _stddev(nullptr), _global_sum(nullptr), _global_sum_squared(nullptr), _mtx() + : _input(nullptr), _mean(nullptr), _stddev(nullptr), _global_sum(nullptr), _global_sum_squared(nullptr), _mtx(), _border_size(0) { } +BorderSize NEMeanStdDevKernel::border_size() const +{ + return _border_size; +} + void NEMeanStdDevKernel::configure(const IImage *input, float *mean, uint64_t *global_sum, float *stddev, uint64_t *global_sum_squared) { ARM_COMPUTE_ERROR_ON_TENSOR_NOT_2D(input); @@ -105,6 +110,9 @@ void NEMeanStdDevKernel::configure(const IImage *input, float *mean, uint64_t *g constexpr unsigned int num_elems_processed_per_iteration = 16; + _border_size = BorderSize(std::max(static_cast(num_elems_processed_per_iteration) - static_cast(input->info()->dimension(0)), + static_cast(input->info()->dimension(0) % num_elems_processed_per_iteration))); + // Configure kernel window Window win = calculate_max_window(*input->info(), Steps(num_elems_processed_per_iteration)); -- cgit v1.2.1