diff options
author | Viet-Hoa Do <viet-hoa.do@arm.com> | 2023-08-08 11:03:06 +0100 |
---|---|---|
committer | Viet-Hoa Do <viet-hoa.do@arm.com> | 2023-08-08 13:28:50 +0000 |
commit | 78da34c101d22e13f657e07e56e4baf52c88a016 (patch) | |
tree | 979124cb487a7b72aa890c7d35f6aba56df2c81b /arm_compute/core/utils/misc/ShapeCalculator.h | |
parent | 1df9f6ed4245489b74875893c695367bd0d6e3d8 (diff) | |
download | ComputeLibrary-78da34c101d22e13f657e07e56e4baf52c88a016.tar.gz |
Fix failure in MeanReduce layer
Resolves: COMPMID-6423
Signed-off-by: Viet-Hoa Do <viet-hoa.do@arm.com>
Change-Id: I9cec051a7d1a2956218f8a6d8263bd5424f6d389
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/10072
Reviewed-by: Gunes Bayir <gunes.bayir@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Benchmark: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'arm_compute/core/utils/misc/ShapeCalculator.h')
-rw-r--r-- | arm_compute/core/utils/misc/ShapeCalculator.h | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/arm_compute/core/utils/misc/ShapeCalculator.h b/arm_compute/core/utils/misc/ShapeCalculator.h index 77ad33910b..4c2037ab8d 100644 --- a/arm_compute/core/utils/misc/ShapeCalculator.h +++ b/arm_compute/core/utils/misc/ShapeCalculator.h @@ -64,7 +64,7 @@ inline TensorShape calculate_reduce_mean_shape(ITensorInfo *input, const Coordin std::sort(axis_local.begin(), axis_local.begin() + reduction_ops); for(int i = 0; i < reduction_ops; ++i) { - out_shape.remove_dimension(axis_local[i] - i); + out_shape.remove_dimension(axis_local[i] - i, false); } return out_shape; } |