Function Description

conv1d_?

fp32, fp16

General 1D convolution interface

conv1d_direct_?

fp32, fp16

1D convolution interface implemented by the Direct algorithm

conv1d_fft_?

fp32, fp16

1D convolution interface implemented by the FFT algorithm

conv1d_gemm_?

fp32, fp16

1D convolution interface implemented by the GEMM algorithm

conv1d_winograd_?

fp32, fp16

1D convolution interface implemented by the Winograd algorithm

conv2d_?

fp32, fp16

General 2D convolution interface

conv2d_direct_?

fp32, fp16

2D convolution interface implemented by the Direct algorithm

conv2d_fft_?

fp32, fp16

2D convolution interface implemented by the FFT algorithm

conv2d_gemm_?

fp32, fp16

2D convolution interface implemented by the GEMM algorithm

conv2d_winograd_?

fp32, fp16

2D convolution interface implemented by the Winograd algorithm