diff options
Diffstat (limited to 'arm_compute/runtime/NEON/functions/NEStackLayer.h')
| -rw-r--r-- | arm_compute/runtime/NEON/functions/NEStackLayer.h | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/arm_compute/runtime/NEON/functions/NEStackLayer.h b/arm_compute/runtime/NEON/functions/NEStackLayer.h index e83995619d..9288035060 100644 --- a/arm_compute/runtime/NEON/functions/NEStackLayer.h +++ b/arm_compute/runtime/NEON/functions/NEStackLayer.h @@ -50,7 +50,7 @@ public: * * @note Supported input tensor rank: up to 4 * - * @param[in] input The vectors containing all the tensors with the same shape to stack. Data types supported: U8/S8/QASYMM8/U16/S16/F16/U32/S32/F32 + * @param[in] input The vectors containing all the tensors with the same shape to stack. Data types supported: All * @param[in] axis The dimension to stack the tensors along. It must be smaller than the number of input dimensions. * Negative values wrap around * @param[out] output Output tensor. Data types supported: Same as @p input. @@ -60,7 +60,7 @@ public: * * @note Supported input tensor rank: up to 4 * - * @param[in] input The vectors containing all the tensors info with the same shape to stack. Data types supported: U8/S8/QASYMM8/U16/S16/F16/U32/S32/F32 + * @param[in] input The vectors containing all the tensors info with the same shape to stack. Data types supported: All * @param[in] axis The dimension to stack the tensors along. It must be smaller than the number of input dimensions. * Negative values wrap around * @param[in] output Output tensor info. Data types supported: Same as @p input. |