diff options
author | Giorgio Arena <giorgio.arena@arm.com> | 2020-12-11 13:39:02 +0000 |
---|---|---|
committer | Giorgio Arena <giorgio.arena@arm.com> | 2020-12-11 14:53:45 +0000 |
commit | ec241b48ea7481e797285788fd68e5e1d42382bb (patch) | |
tree | 9fa8354cc5ec6f018ab04adf8ed68612cfd0e9a4 /arm_compute/core/Helpers.h | |
parent | c53266e45f3c8c07dff88c61e5bfa01c6d3ba3f0 (diff) | |
download | ComputeLibrary-ec241b48ea7481e797285788fd68e5e1d42382bb.tar.gz |
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>
Diffstat (limited to 'arm_compute/core/Helpers.h')
-rw-r--r-- | arm_compute/core/Helpers.h | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/arm_compute/core/Helpers.h b/arm_compute/core/Helpers.h index 5a8d6efe9d..b6635aba6d 100644 --- a/arm_compute/core/Helpers.h +++ b/arm_compute/core/Helpers.h @@ -145,7 +145,7 @@ inline void permute(TensorShape &shape, const PermutationVector &perm) 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 } } |