Skip to main content
Log in

Mojo module

conv

Structs

  • ConvDirectNHWC: Implement the outer loops for direct convolution. Collapse N, HO, WO into one dimension n_ho_wo. Tile n_ho_wo, C, and F. The tile factor for C and F are chosen by a heuristic prioritizing C. n_ho_wo is tiled by micro kernel's height.
  • Naive2dConvolution: Struct wrapper for naive 2d convolution implementation.

Functions