Allow TensorShape to increment num_dimensions for new unit dimensions, without correction
Resolves: COMPMID-4053
Change-Id: Ie0b58b393e07518deb2c1fe4f82cbf0ce257f39a
Signed-off-by: Giorgio Arena <giorgio.arena@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/4691
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: SiCong Li <sicong.li@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
diff --git a/arm_compute/core/Helpers.h b/arm_compute/core/Helpers.h
index 5a8d6ef..b6635ab 100644
--- a/arm_compute/core/Helpers.h
+++ b/arm_compute/core/Helpers.h
@@ -145,7 +145,7 @@
for(unsigned int i = 0; i < perm.num_dimensions(); ++i)
{
size_t dimension_val = (perm[i] < shape.num_dimensions()) ? shape_copy[perm[i]] : 1;
- shape.set(i, dimension_val, false); // Avoid changes in _num_dimension
+ shape.set(i, dimension_val, false, false); // Avoid changes in _num_dimension
}
}