diff options
Diffstat (limited to 'arm_compute/runtime/NEON/functions/NEDirectConvolutionLayer.h')
| -rw-r--r-- | arm_compute/runtime/NEON/functions/NEDirectConvolutionLayer.h | 13 |
1 files changed, 6 insertions, 7 deletions
diff --git a/arm_compute/runtime/NEON/functions/NEDirectConvolutionLayer.h b/arm_compute/runtime/NEON/functions/NEDirectConvolutionLayer.h index ae384ffa56..a4a55d10f8 100644 --- a/arm_compute/runtime/NEON/functions/NEDirectConvolutionLayer.h +++ b/arm_compute/runtime/NEON/functions/NEDirectConvolutionLayer.h @@ -54,11 +54,11 @@ public: /** Set the input, weights, biases and output tensors. * * @note: DirectConvolution only works in the following configurations: - * 1x1 convolution with stride_x = 1/2/3, stride_y = 1/2/3 data type = QS8/QS16/F16/F32 - * 3x3 convolution with stride_x = 1/2/3, stride_y = 1/2/3 data type = QS8/F16/F32 + * 1x1 convolution with stride_x = 1/2/3, stride_y = 1/2/3 data type = F16/F32 + * 3x3 convolution with stride_x = 1/2/3, stride_y = 1/2/3 data type = F16/F32 * 5x5 convolution with stride_x = 1/2/3, stride_y = 1/2/3 data type = F32 * - * @param[in, out] input Input tensor. Data types supported: QS8/QS16/F16/F32. + * @param[in, out] input Input tensor. Data types supported: F16/F32. * @param[in] weights Set of kernels to convolve the input volume. * Supported sizes: 1x1, 3x3 and 5x5. * The 3rd dimension must be the same as the input's volume 3rd dimension. @@ -73,11 +73,11 @@ public: /** Static function to check if given info will lead to a valid configuration of @ref NEDirectConvolutionLayer * * @note: DirectConvolution only works in the following configurations: - * 1x1 convolution with stride_x = 1/2/3, stride_y = 1/2/3 data type = QS8/QS16/F16/F32 - * 3x3 convolution with stride_x = 1/2/3, stride_y = 1/2/3 data type = QS8/F16/F32 + * 1x1 convolution with stride_x = 1/2/3, stride_y = 1/2/3 data type = F16/F32 + * 3x3 convolution with stride_x = 1/2/3, stride_y = 1/2/3 data type = F16/F32 * 5x5 convolution with stride_x = 1/2/3, stride_y = 1/2/3 data type = F32 * - * @param[in] input Input tensor. Data types supported: QS8/QS16/F16/F32. + * @param[in] input Input tensor. Data types supported: F16/F32. * @param[in] weights Set of kernels to convolve the input volume. * Supported sizes: 1x1, 3x3 and 5x5. * The 3rd dimension must be the same as the input's volume 3rd dimension. @@ -104,7 +104,6 @@ private: NEActivationLayer _activationlayer_function; Tensor _accumulator; bool _has_bias; - bool _is_fixed_point; bool _is_activationlayer_enabled; unsigned int _dim_split; }; |