Skip to main content

Module definition

Module definition 

Source

Structs§

BackwardDataBlueprint
BackwardWeightBlueprint
Conv2dCost
Minimal representation of a 2D convolution’s cost: the shapes it moves and the element types it moves them in.
DepthwiseCost
Minimal representation of a depthwise convolution’s cost. Each channel has its own filter and contracts over nothing else, so this is not the implicit gemm Conv2dCost describes.
ForwardBlueprint

Enums§

ConvBlueprint
Per-operation comptime blueprint for the convolution kernel family.