From ffd31defdb84d4ca1e24e9248d628c0075767302 Mon Sep 17 00:00:00 2001 From: Pablo Tello Date: Wed, 4 Sep 2019 13:38:14 +0100 Subject: COMPMID-2246: NEBoundingBoxTransform support QASYMM16 Change-Id: I0704f71159a3caec4705779cab2ef38aeb33aaca Signed-off-by: Pablo Tello Reviewed-on: https://review.mlplatform.org/c/1864 Comments-Addressed: Arm Jenkins Tested-by: Arm Jenkins Reviewed-by: Giorgio Arena --- src/core/CL/kernels/CLBoundingBoxTransformKernel.cpp | 4 ++++ 1 file changed, 4 insertions(+) (limited to 'src/core/CL/kernels/CLBoundingBoxTransformKernel.cpp') diff --git a/src/core/CL/kernels/CLBoundingBoxTransformKernel.cpp b/src/core/CL/kernels/CLBoundingBoxTransformKernel.cpp index 08e5cc6b3b..8fc6f82bd6 100644 --- a/src/core/CL/kernels/CLBoundingBoxTransformKernel.cpp +++ b/src/core/CL/kernels/CLBoundingBoxTransformKernel.cpp @@ -59,6 +59,10 @@ Status validate_arguments(const ITensorInfo *boxes, const ITensorInfo *pred_boxe ARM_COMPUTE_RETURN_ERROR_ON(boxes_qinfo.offset != 0); ARM_COMPUTE_RETURN_ERROR_ON_DATA_TYPE_NOT_IN(deltas, DataType::QASYMM8); } + else + { + ARM_COMPUTE_RETURN_ERROR_ON_MISMATCHING_DATA_TYPES(boxes, deltas); + } if(pred_boxes->total_size() > 0) { -- cgit v1.2.1