diff options
Diffstat (limited to 'arm_compute/runtime/NEON/functions/NEFullyConnectedLayer.h')
-rw-r--r-- | arm_compute/runtime/NEON/functions/NEFullyConnectedLayer.h | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/arm_compute/runtime/NEON/functions/NEFullyConnectedLayer.h b/arm_compute/runtime/NEON/functions/NEFullyConnectedLayer.h index 1b3f36d866..58b11744a8 100644 --- a/arm_compute/runtime/NEON/functions/NEFullyConnectedLayer.h +++ b/arm_compute/runtime/NEON/functions/NEFullyConnectedLayer.h @@ -35,7 +35,7 @@ namespace arm_compute { -/** Basic function to reshape the weights of Fully Connected layer with Neon. This function calls the following kernels: +/** Basic function to reshape the weights of Fully Connected layer. This function calls the following kernels: * * @note The fully connected layer accepts "weights" tensors only with 2 dimensions. */ @@ -117,7 +117,7 @@ private: }; } // namespace weights_transformations -/** Basic function to compute a Fully Connected layer on Neon. This function calls the following Neon kernels: +/** Basic function to compute a Fully Connected layer. This function calls the following kernels: * -# @ref NEIm2ColKernel (called when the input comes from a convolutional layer) * -# @ref NEFullyConnectedLayerReshapeWeights (if @p are_weights_reshaped is set to false and transpose_weights is set to true ) (called once) * -# @ref NEGEMMMatrixMultiplyKernel or @ref NEGEMMLowpMatrixMultiplyCore (if quantized asymmetric) |