| commit | 11bfe0e99b3af0edbed1bfd2b89242b3b4e45d72 | [log] [tgz] |
|---|---|---|
| author | Manuel Bottini <manuel.bottini@arm.com> | Wed May 20 15:26:36 2020 +0100 |
| committer | Manuel Bottini <manuel.bottini@arm.com> | Wed May 20 17:39:33 2020 +0000 |
| tree | b30dc6980218b51efa9a9ad66831f790243345fe | |
| parent | d5af1a05582584c30baf13e2370659692caa3516 [diff] [blame] |
COMPMID-3491: ReduceMean not properly validating quantizations Change-Id: Ia2563da7271592c4c55dde7fcecaedda6a46a781 Signed-off-by: Manuel Bottini <manuel.bottini@arm.com> Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/3236 Reviewed-by: Michalis Spyrou <michalis.spyrou@arm.com> Comments-Addressed: Arm Jenkins <bsgcomp@arm.com> Tested-by: Arm Jenkins <bsgcomp@arm.com>
diff --git a/src/runtime/NEON/functions/NEReduceMean.cpp b/src/runtime/NEON/functions/NEReduceMean.cpp index 5c936af..d53ed31 100644 --- a/src/runtime/NEON/functions/NEReduceMean.cpp +++ b/src/runtime/NEON/functions/NEReduceMean.cpp
@@ -89,6 +89,7 @@ } const TensorInfo out_info = input->clone()->set_tensor_shape(out_shape); ARM_COMPUTE_RETURN_ERROR_ON_MISMATCHING_SHAPES(output, &out_info); + ARM_COMPUTE_RETURN_ERROR_ON_MISMATCHING_QUANTIZATION_INFO(input, output); } return Status{}; }