diff options
Diffstat (limited to 'src/core/NEON/kernels/NEWinogradConvolutionLayerKernel.h')
-rw-r--r-- | src/core/NEON/kernels/NEWinogradConvolutionLayerKernel.h | 14 |
1 files changed, 7 insertions, 7 deletions
diff --git a/src/core/NEON/kernels/NEWinogradConvolutionLayerKernel.h b/src/core/NEON/kernels/NEWinogradConvolutionLayerKernel.h index 3583735482..75d257de4b 100644 --- a/src/core/NEON/kernels/NEWinogradConvolutionLayerKernel.h +++ b/src/core/NEON/kernels/NEWinogradConvolutionLayerKernel.h @@ -35,7 +35,7 @@ namespace arm_compute // Forward declarations class ITensor; -/** Interface for the Neon kernel to perform Winograd input transform. */ +/** Interface for the kernel to perform Winograd input transform. */ class INEWinogradLayerTransformInputKernel : public INEKernel { public: @@ -96,7 +96,7 @@ public: } }; -/** Neon kernel to perform Winograd input transform. */ +/** Kernel to perform Winograd input transform. */ template <typename T, int OutputTileRows, int OutputTileCols, int KernelRows, int KernelCols> class NEWinogradLayerTransformInputKernel : public INEWinogradLayerTransformInputKernel { @@ -227,7 +227,7 @@ private: ITensor *_workspace; }; -/** Interface for the Neon kernel to perform Winograd output transform. */ +/** Interface for the kernel to perform Winograd output transform. */ class INEWinogradLayerTransformOutputKernel : public INEKernel { public: @@ -310,7 +310,7 @@ public: } }; -/** Neon kernel to perform Winograd output transform. */ +/** Kernel to perform Winograd output transform. */ template <typename T, int OutputTileRows, int OutputTileCols, int KernelRows, int KernelCols> class NEWinogradLayerTransformOutputKernel : public INEWinogradLayerTransformOutputKernel { @@ -437,7 +437,7 @@ private: int _num_channels; }; -/** Interface for the Neon kernel to perform Winograd weights transform. */ +/** Interface for the kernel to perform Winograd weights transform. */ class INEWinogradLayerTransformWeightsKernel : public INEKernel { public: @@ -495,7 +495,7 @@ public: static Status validate(const ITensorInfo *input, const ITensorInfo *weights); }; -/** Neon kernel to perform Winograd weights transform. */ +/** Kernel to perform Winograd weights transform. */ template <typename T, int OutputTileRows, int OutputTileCols, int KernelRows, int KernelCols> class NEWinogradLayerTransformWeightsKernel final : public INEWinogradLayerTransformWeightsKernel { @@ -577,7 +577,7 @@ private: int _num_input_channels; }; -/** Neon kernel to perform Winograd. */ +/** Kernel to perform Winograd. */ template <typename TIn, typename TOut, int OutputTileRows, int OutputTileCols, int KernelRows, int KernelCols> class NEWinogradLayerConfiguration { |