diff options
| author | morgolock <pablo.tello@arm.com> | 2019-12-31 12:20:47 +0000 |
|---|---|---|
| committer | Michele Di Giorgio <michele.digiorgio@arm.com> | 2020-01-09 11:37:01 +0000 |
| commit | a35980546c00ae1647ce033b061530607a5ad1e4 (patch) | |
| tree | 5eba5c7f8bd4edcd36d682faf35b70d2492b4a99 /src/runtime/NEON/functions/NEElementwiseOperators.cpp | |
| parent | bb123bd6f64444141161562aad06cb406762d47a (diff) | |
| download | ComputeLibrary-a35980546c00ae1647ce033b061530607a5ad1e4.tar.gz | |
COMPMID-2994: Add support QASYMM8_SIGNED in NEElementwiseMax
Change-Id: I8261558384ae028f7f016dfd6715de140f0b6445
Signed-off-by: morgolock <pablo.tello@arm.com>
Reviewed-on: https://review.mlplatform.org/c/2528
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com>
Diffstat (limited to 'src/runtime/NEON/functions/NEElementwiseOperators.cpp')
| -rw-r--r-- | src/runtime/NEON/functions/NEElementwiseOperators.cpp | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/src/runtime/NEON/functions/NEElementwiseOperators.cpp b/src/runtime/NEON/functions/NEElementwiseOperators.cpp index 699363111d..ede8c20428 100644 --- a/src/runtime/NEON/functions/NEElementwiseOperators.cpp +++ b/src/runtime/NEON/functions/NEElementwiseOperators.cpp @@ -23,6 +23,7 @@ */ #include "arm_compute/runtime/NEON/functions/NEElementwiseOperations.h" #include <arm_compute/core/NEON/kernels/NEElementwiseOperationKernel.h> +#include "arm_compute/core/Validate.h" #include "arm_compute/core/ITensor.h" #include "support/ToolchainSupport.h" @@ -40,6 +41,11 @@ void NEElementwiseMax::configure(ITensor *input1, ITensor *input2, ITensor *outp Status NEElementwiseMax::validate(const ITensorInfo *input1, const ITensorInfo *input2, const ITensorInfo *output) { + if( input1->data_type() == DataType::QASYMM8_SIGNED) + { + ARM_COMPUTE_RETURN_ERROR_ON_MISMATCHING_DATA_TYPES(input1, output); + ARM_COMPUTE_RETURN_ERROR_ON_MISMATCHING_QUANTIZATION_INFO(input1, input2, output); + } return NEArithmeticOperationKernel::validate(ArithmeticOperation::MAX, input1, input2, output); } |