aboutsummaryrefslogtreecommitdiff
path: root/src/runtime/NEON/functions/NEReductionOperation.cpp
diff options
context:
space:
mode:
authorSheri Zhang <sheri.zhang@arm.com>2020-09-23 11:22:50 +0100
committerSheri Zhang <sheri.zhang@arm.com>2020-10-06 09:19:36 +0000
commit4d91dc68adf8a4cc07285fe781469231230df3b9 (patch)
tree4b8b53ab30f86921031fd2b6b9ff35dfdecc222b /src/runtime/NEON/functions/NEReductionOperation.cpp
parent47ae441b320c0a9f79f8e6036a0b12a1bf68f9ca (diff)
downloadComputeLibrary-4d91dc68adf8a4cc07285fe781469231230df3b9.tar.gz
COMPMID-3181: Remove padding from NEReductionOperationKernel
COMPMID-3803: Remove padding from NEComplexPixelWiseMultiplicationKernel Signed-off-by: Sheri Zhang <sheri.zhang@arm.com> Change-Id: I309fc4ab62bacbca9203d2680a9d6d52f76f70e6 Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/4078 Comments-Addressed: Arm Jenkins <bsgcomp@arm.com> Tested-by: Arm Jenkins <bsgcomp@arm.com> Reviewed-by: Manuel Bottini <manuel.bottini@arm.com> Reviewed-by: Pablo Marquez <pablo.tello@arm.com>
Diffstat (limited to 'src/runtime/NEON/functions/NEReductionOperation.cpp')
-rw-r--r--src/runtime/NEON/functions/NEReductionOperation.cpp47
1 files changed, 1 insertions, 46 deletions
diff --git a/src/runtime/NEON/functions/NEReductionOperation.cpp b/src/runtime/NEON/functions/NEReductionOperation.cpp
index fc2e6d2912..91176bfa45 100644
--- a/src/runtime/NEON/functions/NEReductionOperation.cpp
+++ b/src/runtime/NEON/functions/NEReductionOperation.cpp
@@ -54,7 +54,7 @@ size_t reduction_window_split_dimension(unsigned int axis)
} // namespace
NEReductionOperation::NEReductionOperation(std::shared_ptr<IMemoryManager> memory_manager)
- : _memory_group(memory_manager), _reduction_kernel(), _fill_border_kernel(), _reshape(), _output_internal(), _window_split(0), _reduction_axis(), _is_reshape_required(false)
+ : _memory_group(memory_manager), _reduction_kernel(), _reshape(), _output_internal(), _window_split(0), _reduction_axis(), _is_reshape_required(false)
{
}
@@ -128,47 +128,6 @@ void NEReductionOperation::configure(ITensor *input, ITensor *output, unsigned i
_window_split = reduction_window_split_dimension(axis);
_reduction_axis = axis;
- if(axis == 0)
- {
- // Configure fill border kernel
- const BorderSize fill_border_size = _reduction_kernel.border_size();
- PixelValue pixelValue;
- switch(op)
- {
- case ReductionOperation::PROD:
- {
- pixelValue = PixelValue(1, input->info()->data_type(), input->info()->quantization_info());
- break;
- }
- case ReductionOperation::MIN:
- {
- pixelValue = std::get<1>(get_min_max(input->info()->data_type()));
- break;
- }
- case ReductionOperation::MAX:
- {
- pixelValue = std::get<0>(get_min_max(input->info()->data_type()));
- break;
- }
- case ReductionOperation::ARG_IDX_MAX:
- case ReductionOperation::ARG_IDX_MIN:
- {
- pixelValue = PixelValue(0, input->info()->data_type(), input->info()->quantization_info());
- break;
- }
- case ReductionOperation::MEAN_SUM:
- case ReductionOperation::SUM_SQUARE:
- case ReductionOperation::SUM:
- {
- pixelValue = PixelValue(static_cast<uint32_t>(0));
- break;
- }
- default:
- ARM_COMPUTE_ERROR("Reduction Operation unsupported");
- }
- _fill_border_kernel.configure(input, fill_border_size, (is_arg_min_max ? BorderMode::REPLICATE : BorderMode::CONSTANT), pixelValue);
- }
-
if(_is_reshape_required)
{
_reshape.configure(output_internal, output);
@@ -179,10 +138,6 @@ void NEReductionOperation::configure(ITensor *input, ITensor *output, unsigned i
void NEReductionOperation::run()
{
MemoryGroupResourceScope scope_mg(_memory_group);
- if(_reduction_axis == 0)
- {
- NEScheduler::get().schedule(&_fill_border_kernel, Window::DimY);
- }
NEScheduler::get().schedule(&_reduction_kernel, _window_split);
if(_is_reshape_required)
{