From a35980546c00ae1647ce033b061530607a5ad1e4 Mon Sep 17 00:00:00 2001 From: morgolock Date: Tue, 31 Dec 2019 12:20:47 +0000 Subject: COMPMID-2994: Add support QASYMM8_SIGNED in NEElementwiseMax Change-Id: I8261558384ae028f7f016dfd6715de140f0b6445 Signed-off-by: morgolock Reviewed-on: https://review.mlplatform.org/c/2528 Comments-Addressed: Arm Jenkins Tested-by: Arm Jenkins Reviewed-by: Michele Di Giorgio --- src/runtime/NEON/functions/NEElementwiseOperators.cpp | 6 ++++++ 1 file changed, 6 insertions(+) (limited to 'src/runtime/NEON/functions/NEElementwiseOperators.cpp') 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 +#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); } -- cgit v1.2.1