From 089695f0d4b1ebd1bc76ba95e415bce1297808be Mon Sep 17 00:00:00 2001 From: Isabella Gottardi Date: Wed, 17 Oct 2018 18:04:15 +0100 Subject: COMPMID-1659: (Nightly) CLGEMMConvolutionLayer QASYMM8 TensorShape error Change-Id: Ib4ca28b82bd82f0ed4d2c906185d3f4010246616 Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/153986 Reviewed-by: Giuseppe Rossini Tested-by: bsgcomp --- arm_compute/core/utils/misc/ShapeCalculator.h | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) (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 11d20c919f..56f65d0ba8 100644 --- a/arm_compute/core/utils/misc/ShapeCalculator.h +++ b/arm_compute/core/utils/misc/ShapeCalculator.h @@ -113,7 +113,15 @@ inline TensorShape compute_interleaved_shape(const ITensorInfo &a, int mult_inte const int M = a.dimension(1) * a.dimension(2); const int height = std::ceil(M / static_cast(interleave_width)); shape_interleaved_a.set(1, height); - shape_interleaved_a.remove_dimension(2); + + // When the data format is NHWC and the shapes are Nx1x1 + // the tensor shape num_dimensions is automatically set to 1 instead of 3. + // To avoid failures by removing a dimension that doesn't exist + // check if the number of dimensions is greater than 2. + if(shape_interleaved_a.num_dimensions() > 2) + { + shape_interleaved_a.remove_dimension(2); + } } else { -- cgit v1.2.1