diff options
Diffstat (limited to 'src/core/NEON')
-rw-r--r-- | src/core/NEON/kernels/NEReshapeLayerKernel.cpp | 2 | ||||
-rw-r--r-- | src/core/NEON/kernels/NEWinogradConvolutionLayerKernel.cpp | 2 |
2 files changed, 4 insertions, 0 deletions
diff --git a/src/core/NEON/kernels/NEReshapeLayerKernel.cpp b/src/core/NEON/kernels/NEReshapeLayerKernel.cpp index 298f50b93d..c71899144d 100644 --- a/src/core/NEON/kernels/NEReshapeLayerKernel.cpp +++ b/src/core/NEON/kernels/NEReshapeLayerKernel.cpp @@ -35,6 +35,7 @@ #include <cstdint> +/** [NEReshapeLayerKernel Kernel] **/ using namespace arm_compute; namespace @@ -120,3 +121,4 @@ void NEReshapeLayerKernel::run(const Window &window, const ThreadInfo &info) ARM_COMPUTE_ERROR("Unsupported data type!"); } } +/** [NEReshapeLayerKernel Kernel] **/ diff --git a/src/core/NEON/kernels/NEWinogradConvolutionLayerKernel.cpp b/src/core/NEON/kernels/NEWinogradConvolutionLayerKernel.cpp index f5609b6f5c..3e76a080fd 100644 --- a/src/core/NEON/kernels/NEWinogradConvolutionLayerKernel.cpp +++ b/src/core/NEON/kernels/NEWinogradConvolutionLayerKernel.cpp @@ -249,6 +249,7 @@ int NEWinogradLayerTransformWeightsKernel<T, OutputTileRows, OutputTileCols, Ker return WinogradConv::get_kernel_matrix_stride(kernel_shape); } +#ifndef DOXYGEN_SKIP_THIS template <typename T, int OutputTileRows, int OutputTileCols, int KernelRows, int KernelCols> void NEWinogradLayerTransformWeightsKernel<T, OutputTileRows, OutputTileCols, KernelRows, KernelCols>::configure( const ITensor *weights_hwio, @@ -270,6 +271,7 @@ void NEWinogradLayerTransformWeightsKernel<T, OutputTileRows, OutputTileCols, Ke win.set(Window::DimX, Window::Dimension(0, win_last, 1)); INEKernel::configure(win); } +#endif /* DOXYGEN_SKIP_THIS */ template <typename T, int OutputTileRows, int OutputTileCols, int KernelRows, int KernelCols> void NEWinogradLayerTransformWeightsKernel<T, OutputTileRows, OutputTileCols, KernelRows, KernelCols>::run(const Window &window, const ThreadInfo &info) |