diff options
| author | Giorgio Arena <giorgio.arena@arm.com> | 2018-04-20 16:06:21 +0100 |
|---|---|---|
| committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:49:37 +0000 |
| commit | f485a100e3f11911d25a16b4ccc286c6c0816061 (patch) | |
| tree | e5ded0791adee9f8c83279664d20f3db5e490a0e /arm_compute/core/utils | |
| parent | 48c19f1308ecdc7ea37a6bf5ce9459e0954e9007 (diff) | |
| download | ComputeLibrary-f485a100e3f11911d25a16b4ccc286c6c0816061.tar.gz | |
COMPMID-802 Fix NEIm2Col NHWC
Change-Id: I513e0199b6fa665c4a7d2a739f4871b4575ef347
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/128490
Tested-by: Jenkins <bsgcomp@arm.com>
Reviewed-by: Pablo Tello <pablo.tello@arm.com>
Diffstat (limited to 'arm_compute/core/utils')
| -rw-r--r-- | arm_compute/core/utils/misc/ShapeCalculator.h | 6 |
1 files changed, 3 insertions, 3 deletions
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<unsigned int, unsigned int> 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; } |