diff options
author | Viet-Hoa Do <viet-hoa.do@arm.com> | 2023-05-25 12:01:50 +0100 |
---|---|---|
committer | Viet-Hoa Do <viet-hoa.do@arm.com> | 2023-05-31 11:31:12 +0000 |
commit | 545358ea7e65abfba6db5b194264584f1e54cc96 (patch) | |
tree | 09d2cd4a023229ce0c84ac4a1c7e7857ef73155b /arm_compute/core/utils | |
parent | 6c113ed1a95a08d17c2d556bd7b03c901512a34f (diff) | |
download | ComputeLibrary-545358ea7e65abfba6db5b194264584f1e54cc96.tar.gz |
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>
Diffstat (limited to 'arm_compute/core/utils')
-rw-r--r-- | arm_compute/core/utils/misc/ShapeCalculator.h | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/arm_compute/core/utils/misc/ShapeCalculator.h b/arm_compute/core/utils/misc/ShapeCalculator.h index f9352650b6..3461d6a341 100644 --- a/arm_compute/core/utils/misc/ShapeCalculator.h +++ b/arm_compute/core/utils/misc/ShapeCalculator.h @@ -404,8 +404,8 @@ inline TensorShape compute_transposed_shape(const ITensorInfo &input) { 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; } |