aboutsummaryrefslogtreecommitdiff
path: root/arm_compute/runtime/NEON/functions/NEQuantizationLayer.h
diff options
context:
space:
mode:
authorManuel Bottini <manuel.bottini@arm.com>2020-02-07 16:31:59 +0000
committerGeorgios Pinitas <georgios.pinitas@arm.com>2020-02-26 11:06:52 +0000
commit4370cffc7fb0da7fb486b9d06d24e16169521876 (patch)
tree3f1ff71e631e3e14efc423a9fb3a4cf9b4b93b94 /arm_compute/runtime/NEON/functions/NEQuantizationLayer.h
parent12f2b8c316155660f1e612fe7e8fab7861decc03 (diff)
downloadComputeLibrary-4370cffc7fb0da7fb486b9d06d24e16169521876.tar.gz
COMPMID-3034: Add NERequantizationLayerKernel
Change-Id: I3f098c3c2c2031d8cbe7326eab88a4e78bda867f Signed-off-by: Manuel Bottini <manuel.bottini@arm.com> Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/2704 Comments-Addressed: Arm Jenkins <bsgcomp@arm.com> Tested-by: Arm Jenkins <bsgcomp@arm.com> Reviewed-by: Sang-Hoon Park <sang-hoon.park@arm.com>
Diffstat (limited to 'arm_compute/runtime/NEON/functions/NEQuantizationLayer.h')
-rw-r--r--arm_compute/runtime/NEON/functions/NEQuantizationLayer.h10
1 files changed, 5 insertions, 5 deletions
diff --git a/arm_compute/runtime/NEON/functions/NEQuantizationLayer.h b/arm_compute/runtime/NEON/functions/NEQuantizationLayer.h
index 1cf83e87ad..fc317be81e 100644
--- a/arm_compute/runtime/NEON/functions/NEQuantizationLayer.h
+++ b/arm_compute/runtime/NEON/functions/NEQuantizationLayer.h
@@ -1,5 +1,5 @@
/*
- * Copyright (c) 2017-2019 ARM Limited.
+ * Copyright (c) 2017-2020 ARM Limited.
*
* SPDX-License-Identifier: MIT
*
@@ -48,14 +48,14 @@ public:
NEQuantizationLayer() = default;
/** Set the input and output tensors.
*
- * @param[in] input Source tensor. The dimensions over the third will be interpreted as batches. Data types supported: F32/F16.
- * @param[out] output Destination tensor with the same dimensions of input. Data types supported: QASYMM8/QASYMM16
+ * @param[in] input Source tensor. The dimensions over the third will be interpreted as batches. Data types supported: QASYMM8/QASYMM8_SIGNED/F32/F16.
+ * @param[out] output Destination tensor with the same dimensions of input. Data types supported: QASYMM8/QASYMM8_SIGNED/QASYMM16
*/
void configure(const ITensor *input, ITensor *output);
/** Static function to check if given info will lead to a valid configuration of @ref NEQuantizationLayer
*
- * @param[in] input Input tensor info. The dimensions over the third will be interpreted as batches. Data types supported: F32/F16.
- * @param[in] output Output tensor info. Data types supported: QASYMM8/QASYMM16
+ * @param[in] input Input tensor info. The dimensions over the third will be interpreted as batches. Data types supported: QASYMM8/QASYMM8_SIGNED/F32/F16.
+ * @param[in] output Output tensor info. Data types supported: QASYMM8/QASYMM8_SIGNED/QASYMM16
*
* @return a status
*/