diff options
author | Georgios Pinitas <georgios.pinitas@arm.com> | 2020-06-08 13:44:49 +0100 |
---|---|---|
committer | Georgios Pinitas <georgios.pinitas@arm.com> | 2020-06-08 16:41:49 +0000 |
commit | ebfb2f8701c131294b3c5c1e36547fa3658d09dd (patch) | |
tree | 7f529b0168a2059aee88b44544efa5a7f818fe1e | |
parent | f025a77e99224798cf4a28d7e17aef86d2f5792f (diff) | |
download | ComputeLibrary-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.cpp | 8 |
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; |