diff options
author | Manuel Bottini <manuel.bottini@arm.com> | 2020-05-20 15:26:36 +0100 |
---|---|---|
committer | Manuel Bottini <manuel.bottini@arm.com> | 2020-05-20 17:39:33 +0000 |
commit | 11bfe0e99b3af0edbed1bfd2b89242b3b4e45d72 (patch) | |
tree | b30dc6980218b51efa9a9ad66831f790243345fe /src/runtime/CL | |
parent | d5af1a05582584c30baf13e2370659692caa3516 (diff) | |
download | ComputeLibrary-11bfe0e99b3af0edbed1bfd2b89242b3b4e45d72.tar.gz |
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>
Diffstat (limited to 'src/runtime/CL')
-rw-r--r-- | src/runtime/CL/functions/CLReduceMean.cpp | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/src/runtime/CL/functions/CLReduceMean.cpp b/src/runtime/CL/functions/CLReduceMean.cpp index 3ddaa00d4b..ce447636ec 100644 --- a/src/runtime/CL/functions/CLReduceMean.cpp +++ b/src/runtime/CL/functions/CLReduceMean.cpp @@ -83,6 +83,7 @@ Status validate_config(const ITensorInfo *input, const Coordinates &reduction_ax } 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{}; } |