Module conv

Module conv 

Source
Expand description

Optimized convolution operations

Implements multiple convolution algorithms:

  1. im2col + GEMM (industry standard, 5-10x faster)
  2. Winograd (for 3x3 kernels, 2-4x faster)
  3. Direct convolution (fallback)

Functionsยง

conv2d_optimized
Optimized 2D convolution