diff options
author | Vidhya Sudhan Loganathan <vidhyasudhan.loganathan@arm.com> | 2019-06-28 14:09:53 +0100 |
---|---|---|
committer | VidhyaSudhan Loganathan <vidhyasudhan.loganathan@arm.com> | 2019-06-28 14:15:30 +0000 |
commit | 338595bca8ab60492f10626860acb1ab3722b1ce (patch) | |
tree | 03504ec3a2973e30c80f9bf56b77b4a4c7c9d83c /arm_compute/runtime/NEON/functions/NEConcatenateLayer.h | |
parent | 7026b303d636e7639f8877ae8d5eff54f39c1121 (diff) | |
download | ComputeLibrary-338595bca8ab60492f10626860acb1ab3722b1ce.tar.gz |
COMPMID-2234 : Add support for axis 3 in NE/CLConcatenateLayer
Change-Id: Ic86f89ece3afe72809bc69c6de6fee7d21daa1d4
Signed-off-by: Vidhya Sudhan Loganathan <vidhyasudhan.loganathan@arm.com>
Reviewed-on: https://review.mlplatform.org/c/1440
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Gian Marco Iodice <gianmarco.iodice@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'arm_compute/runtime/NEON/functions/NEConcatenateLayer.h')
-rw-r--r-- | arm_compute/runtime/NEON/functions/NEConcatenateLayer.h | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/arm_compute/runtime/NEON/functions/NEConcatenateLayer.h b/arm_compute/runtime/NEON/functions/NEConcatenateLayer.h index 8c97efc4f0..953e3fa641 100644 --- a/arm_compute/runtime/NEON/functions/NEConcatenateLayer.h +++ b/arm_compute/runtime/NEON/functions/NEConcatenateLayer.h @@ -45,6 +45,7 @@ class Status; * -# @ref NEWidthConcatenateLayerKernel (if underlying concatenation axis is 0). * -# @ref NEHeightConcatenateLayerKernel (if underlying concatenation axis is 1). * -# @ref NEDepthConcatenateLayerKernel (if underlying concatenation axis is 2). + * -# @ref NEBatchConcatenateLayerKernel (if underlying concatenation axis is 3). */ class NEConcatenateLayer : public IFunction { @@ -58,7 +59,7 @@ public: * * @param[in,out] inputs_vector The vectors containing all the tensors to concatenate. Data types supported: QASYMM8/F16/F32. * @param[out] output Output tensor. Data types supported: Same as @p input. - * @param[in] axis Concatenation axis. Supported underlying concatenation axis are 0, 1 and 2. + * @param[in] axis Concatenation axis. Supported underlying concatenation axis are 0, 1, 2 and 3. */ void configure(std::vector<ITensor *> inputs_vector, ITensor *output, size_t axis); void configure(std::vector<const ITensor *> inputs_vector, ITensor *output, size_t axis); @@ -69,7 +70,7 @@ public: * * @param[in] inputs_vector The vectors containing all the tensors info to concatenate. Data types supported: QASYMM8/F16/F32. * @param[in] output Output tensor info. Data types supported: Same as @p input. - * @param[in] axis Concatenation axis. Supported underlying concatenation axis are 0, 1 and 2. + * @param[in] axis Concatenation axis. Supported underlying concatenation axis are 0, 1, 2 and 3. * * @return a status */ |