diff options
| author | Pablo Tello <pablo.tello@arm.com> | 2019-11-08 13:47:53 +0000 |
|---|---|---|
| committer | Michele Di Giorgio <michele.digiorgio@arm.com> | 2019-11-22 14:52:40 +0000 |
| commit | 9397515779a021638bc1b997f450c2b1f8e9ad93 (patch) | |
| tree | b2ec42ba8b46d3813d5fee0f61cd54f2320cf8ec /arm_compute/runtime/NEON/functions/NEReduceMean.h | |
| parent | a0a3d20a67f6e5ce42879e0676127dc36721a6d6 (diff) | |
| download | ComputeLibrary-9397515779a021638bc1b997f450c2b1f8e9ad93.tar.gz | |
COMPMID-2855: NEReduceMean throws error for invalid configs
Change-Id: I600507d0de19d7da6c1a13edcfff0a11ea6b5264
Signed-off-by: Pablo Tello <pablo.tello@arm.com>
Reviewed-on: https://review.mlplatform.org/c/2254
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com>
Reviewed-by: Giorgio Arena <giorgio.arena@arm.com>
Reviewed-by: Giuseppe Rossini <giuseppe.rossini@arm.com>
Reviewed-by: Michalis Spyrou <michalis.spyrou@arm.com>
Reviewed-by: Manuel Bottini <manuel.bottini@arm.com>
Reviewed-by: SiCong Li <sicong.li@arm.com>
Diffstat (limited to 'arm_compute/runtime/NEON/functions/NEReduceMean.h')
| -rw-r--r-- | arm_compute/runtime/NEON/functions/NEReduceMean.h | 2 |
1 files changed, 1 insertions, 1 deletions
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<NEReductionOperation> _reduction_kernels; std::vector<Tensor> _reduced_outs; NEReshapeLayer _reshape; - unsigned int _reduction_ops; + int _reduction_ops; bool _keep_dims; }; } // namespace arm_compute |