From a3b4ce2b848442c4f8bb349307e1fc4a164c3a33 Mon Sep 17 00:00:00 2001 From: Anthony Barbier Date: Mon, 9 Oct 2017 11:04:30 +0100 Subject: COMPMID-556: A tensor cannot be of dimension 0 It should always be at least 1. Change-Id: I5b9012a227ccd28c20baf2dce84e8a2be3bd789c Reviewed-on: http://mpd-gerrit.cambridge.arm.com/90746 Tested-by: Kaizen Reviewed-by: Anthony Barbier --- arm_compute/core/TensorShape.h | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'arm_compute') diff --git a/arm_compute/core/TensorShape.h b/arm_compute/core/TensorShape.h index 8d15c50220..3b395e74ce 100644 --- a/arm_compute/core/TensorShape.h +++ b/arm_compute/core/TensorShape.h @@ -164,7 +164,7 @@ private: /** Remove trailing dimensions of size 1 from the reported number of dimensions. */ void apply_dimension_correction() { - for(int i = static_cast(_num_dimensions) - 1; i >= 0; --i) + for(int i = static_cast(_num_dimensions) - 1; i > 0; --i) { if(_id[i] == 1) { -- cgit v1.2.1