diff options
author | Sheri Zhang <sheri.zhang@arm.com> | 2020-09-23 11:22:50 +0100 |
---|---|---|
committer | Sheri Zhang <sheri.zhang@arm.com> | 2020-10-06 09:19:36 +0000 |
commit | 4d91dc68adf8a4cc07285fe781469231230df3b9 (patch) | |
tree | 4b8b53ab30f86921031fd2b6b9ff35dfdecc222b /src/runtime | |
parent | 47ae441b320c0a9f79f8e6036a0b12a1bf68f9ca (diff) | |
download | ComputeLibrary-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')
-rw-r--r-- | src/runtime/NEON/functions/NEReductionOperation.cpp | 47 |
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) { |