diff options
| author | Vidhya Sudhan Loganathan <vidhyasudhan.loganathan@arm.com> | 2018-11-19 15:18:20 +0000 |
|---|---|---|
| committer | Anthony Barbier <Anthony.barbier@arm.com> | 2018-11-19 16:23:00 +0000 |
| commit | d646ae17893a89771cdc61ac8790f7e241ecd799 (patch) | |
| tree | 753ec98326b58064f520464feb112142a779534d /src/core/NEON | |
| parent | e27887f9ec5e71b89829c7e8735839249e14bff9 (diff) | |
| download | ComputeLibrary-d646ae17893a89771cdc61ac8790f7e241ecd799.tar.gz | |
COMPMID-1065 : Create documentation explaining how to add new functions / kernels
Change-Id: I98183f95814442b6f3dbb67a1bdae99df05b9b01
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) |