From ab0a77edcb9f48de2aad216323b791d0dd95a3cd Mon Sep 17 00:00:00 2001 From: Michele Di Giorgio Date: Wed, 21 Jun 2017 15:36:24 +0100 Subject: COMPMID-409: Add support for QS8 and QS16 CLPixelWiseMultiplication. Change-Id: I7f66d49d746ba9fb6e726ccab83d3a97b8ddef80 Reviewed-on: http://mpd-gerrit.cambridge.arm.com/78491 Reviewed-by: Georgios Pinitas Tested-by: Kaizen --- src/core/NEON/kernels/NEPixelWiseMultiplicationKernel.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'src/core/NEON/kernels/NEPixelWiseMultiplicationKernel.cpp') diff --git a/src/core/NEON/kernels/NEPixelWiseMultiplicationKernel.cpp b/src/core/NEON/kernels/NEPixelWiseMultiplicationKernel.cpp index 7c9514723b..c3f61ac94a 100644 --- a/src/core/NEON/kernels/NEPixelWiseMultiplicationKernel.cpp +++ b/src/core/NEON/kernels/NEPixelWiseMultiplicationKernel.cpp @@ -360,7 +360,7 @@ void NEPixelWiseMultiplicationKernel::configure(const ITensor *input1, const ITe ARM_COMPUTE_ERROR_ON_DATA_TYPE_CHANNEL_NOT_IN(output, 1, DataType::U8, DataType::QS8, DataType::S16, DataType::F32); ARM_COMPUTE_ERROR_ON_MSG(output->info()->data_type() == DataType::U8 && (input1->info()->data_type() != DataType::U8 || input2->info()->data_type() != DataType::U8), "Output can only be U8 if both inputs are U8"); - if(output->info()->data_type() == DataType::QS8 || input1->info()->data_type() == DataType::QS8 || output->info()->data_type() == DataType::QS8) + if(input1->info()->data_type() == DataType::QS8) { // All data types must be QS8 ARM_COMPUTE_ERROR_ON_MISMATCHING_DATA_TYPES(input1, input2, output); -- cgit v1.2.1