From f485a100e3f11911d25a16b4ccc286c6c0816061 Mon Sep 17 00:00:00 2001 From: Giorgio Arena Date: Fri, 20 Apr 2018 16:06:21 +0100 Subject: COMPMID-802 Fix NEIm2Col NHWC Change-Id: I513e0199b6fa665c4a7d2a739f4871b4575ef347 Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/128490 Tested-by: Jenkins Reviewed-by: Pablo Tello --- arm_compute/core/utils/misc/ShapeCalculator.h | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'arm_compute/core/utils') diff --git a/arm_compute/core/utils/misc/ShapeCalculator.h b/arm_compute/core/utils/misc/ShapeCalculator.h index b91e52a657..8d4c024f62 100644 --- a/arm_compute/core/utils/misc/ShapeCalculator.h +++ b/arm_compute/core/utils/misc/ShapeCalculator.h @@ -168,9 +168,9 @@ inline TensorShape compute_im2col_conv_shape(const ITensorInfo *input, const Siz const int channel_idx = get_data_layout_dimension_index(data_layout, DataLayoutDimension::CHANNEL); std::pair out_dims = scaled_dimensions(output_shape[width_idx], output_shape[height_idx], kernel_dims.width, kernel_dims.height, conv_info, dilation); - output_shape.set(width_idx, (output_shape[channel_idx] * kernel_dims.area() + (has_bias ? 1 : 0))); - output_shape.set(height_idx, (out_dims.first * out_dims.second)); - output_shape.set(channel_idx, 1); + output_shape.set(0, (output_shape[channel_idx] * kernel_dims.area() + (has_bias ? 1 : 0))); + output_shape.set(1, (out_dims.first * out_dims.second)); + output_shape.set(2, 1); return output_shape; } -- cgit v1.2.1