diff options
author | Manuel Bottini <manuel.bottini@arm.com> | 2021-07-01 18:13:33 +0100 |
---|---|---|
committer | Georgios Pinitas <georgios.pinitas@arm.com> | 2021-07-09 09:42:34 +0000 |
commit | 24b892072a2bd8190ba63d09fb0082113d7d032a (patch) | |
tree | 6376c4449f0d2d7f146496ab32604b3835acf976 /arm_compute/runtime/NEON | |
parent | da816752cad76c8e1b367e8e9c648994a1af599a (diff) | |
download | ComputeLibrary-24b892072a2bd8190ba63d09fb0082113d7d032a.tar.gz |
Port NECol2ImKernel
Resolves: COMPMID-4511
Change-Id: Id6335cb23ef22bba02083498025da0ecb1647714
Signed-off-by: Manuel Bottini <manuel.bottini@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/5898
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'arm_compute/runtime/NEON')
-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; |