diff options
Diffstat (limited to 'arm_compute/runtime/NEON/functions/NEGEMMConvolutionLayer.h')
-rw-r--r-- | arm_compute/runtime/NEON/functions/NEGEMMConvolutionLayer.h | 8 |
1 files changed, 4 insertions, 4 deletions
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<cpu::kernels::CpuIm2ColKernel> _im2col_kernel; NEGEMM _mm_gemm; NEGEMMLowpMatrixMultiplyCore _mm_gemmlowp; - std::unique_ptr<NECol2ImKernel> _col2im_kernel; + std::unique_ptr<cpu::kernels::CpuCol2ImKernel> _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; |