diff options
Diffstat (limited to 'arm_compute/runtime/NEON/functions/NEDetectionPostProcessLayer.h')
-rw-r--r-- | arm_compute/runtime/NEON/functions/NEDetectionPostProcessLayer.h | 12 |
1 files changed, 11 insertions, 1 deletions
diff --git a/arm_compute/runtime/NEON/functions/NEDetectionPostProcessLayer.h b/arm_compute/runtime/NEON/functions/NEDetectionPostProcessLayer.h index d5c1f0ab6f..2affa8d49e 100644 --- a/arm_compute/runtime/NEON/functions/NEDetectionPostProcessLayer.h +++ b/arm_compute/runtime/NEON/functions/NEDetectionPostProcessLayer.h @@ -1,5 +1,5 @@ /* - * Copyright (c) 2019-2020 Arm Limited. + * Copyright (c) 2019-2021 Arm Limited. * * SPDX-License-Identifier: MIT * @@ -57,6 +57,16 @@ public: ~NEDetectionPostProcessLayer() = default; /** Configure the detection output layer NE function * + * Valid data layouts: + * - All + * + * Valid data type configurations: + * |src0 - src2 |dst0 - dst3 | + * |:--------------|:--------------| + * |QASYMM8 |F32 | + * |QASYMM8_SIGNED |F32 | + * |F32 |F32 | + * * @param[in] input_box_encoding The bounding box input tensor. Data types supported: QASYMM8/QASYMM8_SIGNED/F32. * @param[in] input_score The class prediction input tensor. Data types supported: same as @p input_box_encoding. * @param[in] input_anchors The anchors input tensor. Data types supported: same as @p input_box_encoding. |