diff options
author | Manuel Bottini <manuel.bottini@arm.com> | 2021-07-06 15:01:35 +0100 |
---|---|---|
committer | Manuel Bottini <manuel.bottini@arm.com> | 2021-07-15 16:27:56 +0000 |
commit | 29599d01a8f498e33b9c6995bd879473dc02e077 (patch) | |
tree | 2dcbd399c5cf0a87edbd1885e29e8cb1ed1ce9bc /src/core/cpu/kernels/CpuIm2ColKernel.cpp | |
parent | 13ef1763d6eef9606afaed90fb258d1a4577f15b (diff) | |
download | ComputeLibrary-29599d01a8f498e33b9c6995bd879473dc02e077.tar.gz |
Port NEGEMMConvolutionLayer
Details:
port NEWeightsReshapeKernel to CpuWeightsReshapeKernel
port NEGEMMConvolutionLayer to CpuGEMMConvolutionLayer
Resolves: COMPMID-4509
Change-Id: I3c7051e2c3f6d808a7ccb898aad70e5b221b9dc3
Signed-off-by: Manuel Bottini <manuel.bottini@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/5938
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com>
Diffstat (limited to 'src/core/cpu/kernels/CpuIm2ColKernel.cpp')
-rw-r--r-- | src/core/cpu/kernels/CpuIm2ColKernel.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/core/cpu/kernels/CpuIm2ColKernel.cpp b/src/core/cpu/kernels/CpuIm2ColKernel.cpp index a5dbcc29c8..ca6c9bfab4 100644 --- a/src/core/cpu/kernels/CpuIm2ColKernel.cpp +++ b/src/core/cpu/kernels/CpuIm2ColKernel.cpp @@ -331,7 +331,7 @@ void CpuIm2ColKernel::run_im2col(const ITensor *src, ITensor *dst, const Window in, out); } -void CpuIm2ColKernel::configure(ITensorInfo *src, ITensorInfo *dst, const Size2D &kernel_dims, const PadStrideInfo &conv_info, +void CpuIm2ColKernel::configure(const ITensorInfo *src, ITensorInfo *dst, const Size2D &kernel_dims, const PadStrideInfo &conv_info, bool has_bias, const Size2D &dilation, unsigned int num_groups) { ARM_COMPUTE_ERROR_ON_NULLPTR(src, dst); |