From 24b892072a2bd8190ba63d09fb0082113d7d032a Mon Sep 17 00:00:00 2001 From: Manuel Bottini Date: Thu, 1 Jul 2021 18:13:33 +0100 Subject: Port NECol2ImKernel Resolves: COMPMID-4511 Change-Id: Id6335cb23ef22bba02083498025da0ecb1647714 Signed-off-by: Manuel Bottini Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/5898 Tested-by: Arm Jenkins Reviewed-by: Georgios Pinitas Comments-Addressed: Arm Jenkins --- arm_compute/runtime/NEON/functions/NEGEMMConvolutionLayer.h | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) (limited to 'arm_compute/runtime/NEON/functions/NEGEMMConvolutionLayer.h') diff --git a/arm_compute/runtime/NEON/functions/NEGEMMConvolutionLayer.h b/arm_compute/runtime/NEON/functions/NEGEMMConvolutionLayer.h index 655d733bd1..e3b7d91187 100644 --- a/arm_compute/runtime/NEON/functions/NEGEMMConvolutionLayer.h +++ b/arm_compute/runtime/NEON/functions/NEGEMMConvolutionLayer.h @@ -40,13 +40,13 @@ namespace arm_compute { class ITensor; -class NECol2ImKernel; class NEWeightsReshapeKernel; namespace cpu { namespace kernels { class CpuIm2ColKernel; +class CpuCol2ImKernel; } // namespace kernels } // namespace cpu @@ -163,7 +163,7 @@ private: * -# @ref NEGEMMLowpMatrixMultiplyCore (if the data type is QASYMM8/QASYMM8_SIGNED) * -# @ref NEGEMMLowpOutputStage (if the data type is QASYMM8/QASYMM8_SIGNED) * -# @ref NEArithmeticAddition (if biases != nullptr and we have a 1x1 convolution with the NHWC data layout) - * -# @ref NECol2ImKernel (if NCHW data layout) + * -# @ref cpu::kernels::CpuCol2ImKernel (if NCHW data layout) * */ class NEGEMMConvolutionLayer : public IFunction @@ -292,12 +292,12 @@ private: std::unique_ptr _im2col_kernel; NEGEMM _mm_gemm; NEGEMMLowpMatrixMultiplyCore _mm_gemmlowp; - std::unique_ptr _col2im_kernel; + std::unique_ptr _col2im_kernel; NEReshapeLayer _reshape_layer; const ITensor *_input; const ITensor *_original_weights; - const ITensor *_original_output; + ITensor *_original_output; Tensor _im2col_output; Tensor _weights_reshaped; -- cgit v1.2.1