commit | 545358ea7e65abfba6db5b194264584f1e54cc96 | [log] [tgz] |
---|---|---|
author | Viet-Hoa Do <viet-hoa.do@arm.com> | Thu May 25 12:01:50 2023 +0100 |
committer | Viet-Hoa Do <viet-hoa.do@arm.com> | Wed May 31 11:31:12 2023 +0000 |
tree | 09d2cd4a023229ce0c84ac4a1c7e7857ef73155b | |
parent | 6c113ed1a95a08d17c2d556bd7b03c901512a34f [diff] |
Fix number of dimensions changed after transpose Partially resolves: IVGCVSW-7759 Signed-off-by: Viet-Hoa Do <viet-hoa.do@arm.com> Change-Id: I08a3e34db2c3980709718c5a8fc8783dd467d0fe Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/9705 Reviewed-by: Jakub Sujak <jakub.sujak@arm.com> Benchmark: Arm Jenkins <bsgcomp@arm.com> Tested-by: Arm Jenkins <bsgcomp@arm.com>
diff --git a/arm_compute/core/utils/misc/ShapeCalculator.h b/arm_compute/core/utils/misc/ShapeCalculator.h index f935265..3461d6a 100644 --- a/arm_compute/core/utils/misc/ShapeCalculator.h +++ b/arm_compute/core/utils/misc/ShapeCalculator.h
@@ -404,8 +404,8 @@ { TensorShape shape_transposed{ input.tensor_shape() }; - shape_transposed.set(0, input.dimension(1)); - shape_transposed.set(1, input.dimension(0)); + shape_transposed.set(0, input.dimension(1), false); + shape_transposed.set(1, input.dimension(0), false); return shape_transposed; }