From 70ad61972d4e7b5ff69e9f3b2924de0df462e6ee Mon Sep 17 00:00:00 2001 From: Michele Di Giorgio Date: Fri, 6 Sep 2019 17:51:37 +0100 Subject: COMPMID-2635: Add support for QASYMM8 in CPPBoxWithNonMaximaSuppressionLimit Change-Id: Ife35cf865e6573ff7f921eb0b39af89dbf0f5dda Signed-off-by: Michele Di Giorgio Reviewed-on: https://review.mlplatform.org/c/1873 Tested-by: Arm Jenkins Reviewed-by: Pablo Marquez Comments-Addressed: Arm Jenkins --- src/core/CPP/kernels/CPPBoxWithNonMaximaSuppressionLimitKernel.cpp | 1 + 1 file changed, 1 insertion(+) (limited to 'src/core/CPP/kernels/CPPBoxWithNonMaximaSuppressionLimitKernel.cpp') diff --git a/src/core/CPP/kernels/CPPBoxWithNonMaximaSuppressionLimitKernel.cpp b/src/core/CPP/kernels/CPPBoxWithNonMaximaSuppressionLimitKernel.cpp index 02150ff275..62568b4b45 100644 --- a/src/core/CPP/kernels/CPPBoxWithNonMaximaSuppressionLimitKernel.cpp +++ b/src/core/CPP/kernels/CPPBoxWithNonMaximaSuppressionLimitKernel.cpp @@ -351,6 +351,7 @@ void CPPBoxWithNonMaximaSuppressionLimitKernel::configure(const ITensor *scores_ { ARM_COMPUTE_ERROR_ON_NULLPTR(scores_in, boxes_in, scores_out, boxes_out, classes); ARM_COMPUTE_ERROR_ON_DATA_TYPE_CHANNEL_NOT_IN(scores_in, 1, DataType::F16, DataType::F32); + ARM_COMPUTE_ERROR_ON_MISMATCHING_DATA_TYPES(scores_in, boxes_in, scores_out); const unsigned int num_classes = scores_in->info()->dimension(0); ARM_COMPUTE_UNUSED(num_classes); -- cgit v1.2.1