diff options
| author | Pablo Tello <pablo.tello@arm.com> | 2017-07-06 16:43:14 +0100 |
|---|---|---|
| committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-09-17 14:16:42 +0100 |
| commit | 0d176141ca759f0f45b47ed32547f1e44fd875fb (patch) | |
| tree | a6c92a40567347e731d9d5efbe2a644bf4afb94b /arm_compute/runtime/NEON/functions/NEDirectConvolutionLayer.h | |
| parent | b7c2a99f847d3baef1710be5cf34f978514101dd (diff) | |
| download | ComputeLibrary-0d176141ca759f0f45b47ed32547f1e44fd875fb.tar.gz | |
COMPMID-421: Added FP16 support to the NEON Direct Convolution function.
Change-Id: I3a1aa2ce985ecf95fc5f441a6e6d43b4935306ee
Reviewed-on: http://mpd-gerrit.cambridge.arm.com/79965
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
Tested-by: Kaizen <jeremy.johnson+kaizengerrit@arm.com>
Diffstat (limited to 'arm_compute/runtime/NEON/functions/NEDirectConvolutionLayer.h')
| -rw-r--r-- | arm_compute/runtime/NEON/functions/NEDirectConvolutionLayer.h | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/arm_compute/runtime/NEON/functions/NEDirectConvolutionLayer.h b/arm_compute/runtime/NEON/functions/NEDirectConvolutionLayer.h index a356cac7c8..a66cab3013 100644 --- a/arm_compute/runtime/NEON/functions/NEDirectConvolutionLayer.h +++ b/arm_compute/runtime/NEON/functions/NEDirectConvolutionLayer.h @@ -48,7 +48,7 @@ public: NEDirectConvolutionLayer(); /** Set the input, weights, biases and output tensors. * - * @param[in, out] input Input tensor. Data types supported: QS8/F32. + * @param[in, out] input Input tensor. Data types supported: QS8/F16/F32. * @param[in] weights Set of kernels to convolve the input volume. * The 3rd dimension must be the same as the input's volume 3rd dimension. * Data type supported: Same as @p input. |