From 0d176141ca759f0f45b47ed32547f1e44fd875fb Mon Sep 17 00:00:00 2001 From: Pablo Tello Date: Thu, 6 Jul 2017 16:43:14 +0100 Subject: 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 Tested-by: Kaizen --- arm_compute/runtime/NEON/functions/NEDirectConvolutionLayer.h | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'arm_compute/runtime/NEON/functions/NEDirectConvolutionLayer.h') 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. -- cgit v1.2.1