aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorGeorgios Pinitas <georgios.pinitas@arm.com>2020-06-08 13:44:49 +0100
committerGeorgios Pinitas <georgios.pinitas@arm.com>2020-06-08 16:41:49 +0000
commitebfb2f8701c131294b3c5c1e36547fa3658d09dd (patch)
tree7f529b0168a2059aee88b44544efa5a7f818fe1e
parentf025a77e99224798cf4a28d7e17aef86d2f5792f (diff)
downloadComputeLibrary-ebfb2f8701c131294b3c5c1e36547fa3658d09dd.tar.gz
COMPMID-3513: Fix excessive padding in CLHeightConcatenateLayer
Signed-off-by: Georgios Pinitas <georgios.pinitas@arm.com> Change-Id: Ia93685694287f0c4db1558e24033df76521fc190 Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/3312 Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com> Tested-by: Arm Jenkins <bsgcomp@arm.com> Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
-rw-r--r--src/core/CL/kernels/CLHeightConcatenateLayerKernel.cpp8
1 files changed, 4 insertions, 4 deletions
diff --git a/src/core/CL/kernels/CLHeightConcatenateLayerKernel.cpp b/src/core/CL/kernels/CLHeightConcatenateLayerKernel.cpp
index 5c0eb2a606..e7b09644ae 100644
--- a/src/core/CL/kernels/CLHeightConcatenateLayerKernel.cpp
+++ b/src/core/CL/kernels/CLHeightConcatenateLayerKernel.cpp
@@ -44,13 +44,13 @@ namespace arm_compute
{
namespace
{
-std::pair<Status, Window> validate_and_configure_window(ITensorInfo *input, unsigned int height_offset, ITensorInfo *output, unsigned int &num_elems_processed_per_iteration)
+std::pair<Status, Window> validate_and_configure_window(ITensorInfo *input, ITensorInfo *output, unsigned int &num_elems_processed_per_iteration)
{
num_elems_processed_per_iteration = 4;
// The window needs to be based on input as we copy all the heights of input
Window win = calculate_max_window(*input, Steps(num_elems_processed_per_iteration));
AccessWindowHorizontal input_access(input, 0, num_elems_processed_per_iteration);
- AccessWindowHorizontal output_access(output, height_offset, num_elems_processed_per_iteration);
+ AccessWindowHorizontal output_access(output, 0, num_elems_processed_per_iteration);
bool window_changed = update_window_and_padding(win, input_access, output_access);
Window win_collapsed = win.collapse(win, Window::DimZ);
@@ -85,7 +85,7 @@ Status CLHeightConcatenateLayerKernel::validate(const ITensorInfo *input, unsign
{
unsigned int num_elems_processed_per_iteration;
ARM_COMPUTE_RETURN_ON_ERROR(validate_arguments(input, height_offset, output));
- ARM_COMPUTE_RETURN_ON_ERROR(validate_and_configure_window(input->clone().get(), height_offset, output->clone().get(), num_elems_processed_per_iteration).first);
+ ARM_COMPUTE_RETURN_ON_ERROR(validate_and_configure_window(input->clone().get(), output->clone().get(), num_elems_processed_per_iteration).first);
return Status{};
}
@@ -103,7 +103,7 @@ void CLHeightConcatenateLayerKernel::configure(const CLCompileContext &compile_c
_output = output;
_height_offset = height_offset;
- auto win_config = validate_and_configure_window(input->info(), height_offset, output->info(), _num_elems_processed_per_iteration);
+ auto win_config = validate_and_configure_window(input->info(), output->info(), _num_elems_processed_per_iteration);
// Add build options
CLBuildOptions build_opts;