Skip to main content

Module kernels

cubek_convolution

Module kernels 

Source

Modules§

algorithm
Algorithms for the mapped matmul
backward_data
Kernels for data gradients
backward_weight
Kernels for weight gradients
forward
Kernels for forward convolution

Structs§

ConvolutionArgs

Enums§

AcceleratedTileKind
Which tile matmul to use for accelerated algorithms
ReadingStrategy
Which reader to use in simple algorithms
Strategy