diff options
| author | Anthony Barbier <anthony.barbier@arm.com> | 2017-10-09 11:04:30 +0100 |
|---|---|---|
| committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:35:24 +0000 |
| commit | a3b4ce2b848442c4f8bb349307e1fc4a164c3a33 (patch) | |
| tree | 6ce7914b33250f423623b6935aa6ce87afd57b54 | |
| parent | e472082f831815c217677e3f1802ecaae1348e65 (diff) | |
| download | ComputeLibrary-a3b4ce2b848442c4f8bb349307e1fc4a164c3a33.tar.gz | |
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 <jeremy.johnson+kaizengerrit@arm.com>
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
| -rw-r--r-- | arm_compute/core/TensorShape.h | 2 | ||||
| -rw-r--r-- | tests/validation_old/UNIT/TensorShape.cpp | 2 |
2 files changed, 2 insertions, 2 deletions
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<int>(_num_dimensions) - 1; i >= 0; --i) + for(int i = static_cast<int>(_num_dimensions) - 1; i > 0; --i) { if(_id[i] == 1) { diff --git a/tests/validation_old/UNIT/TensorShape.cpp b/tests/validation_old/UNIT/TensorShape.cpp index 3505b1ae05..ef888b24c8 100644 --- a/tests/validation_old/UNIT/TensorShape.cpp +++ b/tests/validation_old/UNIT/TensorShape.cpp @@ -47,7 +47,7 @@ BOOST_DATA_TEST_CASE(Construction, TensorShape{ 2U, 3U, 5U, 7U, 11U }, TensorShape{ 2U, 3U, 5U, 7U, 11U, 13U } }) - ^ boost::unit_test::data::make({ 0, 0, 1, 2, 3, 4, 5, 6 }) ^ boost::unit_test::data::make({ 0, 1, 2, 6, 30, 210, 2310, 30030 }), + ^ boost::unit_test::data::make({ 0, 1, 1, 2, 3, 4, 5, 6 }) ^ boost::unit_test::data::make({ 0, 1, 2, 6, 30, 210, 2310, 30030 }), shape, num_dimensions, total_size) { BOOST_TEST(shape.num_dimensions() == num_dimensions); |