函数说明
函数族 |
数据类型 |
描述 |
|---|---|---|
conv1d_? |
fp32, fp16 |
通用1D卷积接口 |
conv1d_direct_? |
fp32, fp16 |
Direct算法实现的1D卷积接口 |
conv1d_fft_? |
fp32, fp16 |
FFT算法实现的1D卷积接口 |
conv1d_gemm_? |
fp32, fp16 |
GEMM算法实现的1D卷积接口 |
conv1d_winograd_? |
fp32, fp16 |
Winograd算法实现的1D卷积接口 |
conv2d_? |
fp32, fp16 |
通用2D卷积接口 |
conv2d_direct_? |
fp32, fp16 |
Direct算法实现的2D卷积接口 |
conv2d_fft_? |
fp32, fp16 |
FFT算法实现的2D卷积接口 |
conv2d_gemm_? |
fp32, fp16 |
GEMM算法实现的2D卷积接口 |
conv2d_winograd_? |
fp32, fp16 |
Winograd算法实现的2D卷积接口 |
父主题： 通用卷积接口