From 9397515779a021638bc1b997f450c2b1f8e9ad93 Mon Sep 17 00:00:00 2001 From: Pablo Tello Date: Fri, 8 Nov 2019 13:47:53 +0000 Subject: COMPMID-2855: NEReduceMean throws error for invalid configs Change-Id: I600507d0de19d7da6c1a13edcfff0a11ea6b5264 Signed-off-by: Pablo Tello Reviewed-on: https://review.mlplatform.org/c/2254 Comments-Addressed: Arm Jenkins Tested-by: Arm Jenkins Reviewed-by: Michele Di Giorgio Reviewed-by: Giorgio Arena Reviewed-by: Giuseppe Rossini Reviewed-by: Michalis Spyrou Reviewed-by: Manuel Bottini Reviewed-by: SiCong Li --- arm_compute/runtime/NEON/functions/NEReduceMean.h | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'arm_compute/runtime/NEON/functions/NEReduceMean.h') diff --git a/arm_compute/runtime/NEON/functions/NEReduceMean.h b/arm_compute/runtime/NEON/functions/NEReduceMean.h index 9df71f273e..6fbd72b955 100644 --- a/arm_compute/runtime/NEON/functions/NEReduceMean.h +++ b/arm_compute/runtime/NEON/functions/NEReduceMean.h @@ -71,7 +71,7 @@ private: std::vector _reduction_kernels; std::vector _reduced_outs; NEReshapeLayer _reshape; - unsigned int _reduction_ops; + int _reduction_ops; bool _keep_dims; }; } // namespace arm_compute -- cgit v1.2.1