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.h | |
| 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.h')
| -rw-r--r-- | src/core/cpu/kernels/CpuIm2ColKernel.h | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/core/cpu/kernels/CpuIm2ColKernel.h b/src/core/cpu/kernels/CpuIm2ColKernel.h index 4301a237fe..ffac5077b2 100644 --- a/src/core/cpu/kernels/CpuIm2ColKernel.h +++ b/src/core/cpu/kernels/CpuIm2ColKernel.h @@ -77,7 +77,7 @@ public: * @param[in] dilation (Optional) Dilation, in elements, across x and y. Defaults to (1, 1). * @param[in] num_groups (Optional) Number of groups when performing a grouped convolution. num_groups != 1 is not supported */ - void configure(ITensorInfo *src, ITensorInfo *dst, const Size2D &kernel_dims, const PadStrideInfo &conv_info, + void configure(const ITensorInfo *src, ITensorInfo *dst, const Size2D &kernel_dims, const PadStrideInfo &conv_info, bool has_bias, const Size2D &dilation = Size2D(1U, 1U), unsigned int num_groups = 1); /** Static function to check if given info will lead to a valid configuration * |