COMPMID-3271: Add support for QASYMM8_SIGNED in CPPTopKVKernel/CPPTopKV
Signed-off-by: Sheri Zhang <sheri.zhang@arm.com>
Change-Id: Ic34616fc3480ca85cc582e4e3db031d631ed5861
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/2887
Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
diff --git a/arm_compute/runtime/CPP/functions/CPPTopKV.h b/arm_compute/runtime/CPP/functions/CPPTopKV.h
index d41b7d3..c94e277 100644
--- a/arm_compute/runtime/CPP/functions/CPPTopKV.h
+++ b/arm_compute/runtime/CPP/functions/CPPTopKV.h
@@ -1,5 +1,5 @@
/*
- * Copyright (c) 2019 ARM Limited.
+ * Copyright (c) 2019-2020 ARM Limited.
*
* SPDX-License-Identifier: MIT
*
@@ -38,7 +38,7 @@
public:
/** Set the input and output of the kernel.
*
- * @param[in] predictions A batch_size x classes tensor. Data types supported: F16/S32/F32/QASYMM8
+ * @param[in] predictions A batch_size x classes tensor. Data types supported: F16/S32/F32/QASYMM8/QASYMM8_SIGNED
* @param[in] targets A batch_size 1D tensor of class ids. Data types supported: S32
* @param[out] output Computed precision at @p k as a bool 1D tensor. Data types supported: U8
* @param[in] k Number of top elements to look at for computing precision.
@@ -47,7 +47,7 @@
/** Static function to check if given info will lead to a valid configuration of @ref CPPTopKVKernel
*
- * @param[in] predictions A batch_size x classes tensor info. Data types supported: F16/S32/F32/QASYMM8
+ * @param[in] predictions A batch_size x classes tensor info. Data types supported: F16/S32/F32/QASYMM8/QASYMM8_SIGNED
* @param[in] targets A batch_size 1D tensor info of class ids. Data types supported: S32
* @param[in] output Computed precision at @p k as a bool 1D tensor info. Data types supported: U8
* @param[in] k Number of top elements to look at for computing precision.