Module tiling 

Loop tiling and cache blocking.

Computes tiling plans for MatMul and Conv2D operations to improve data locality by partitioning work into cache-friendly tile sizes.

Structs

Conv2DShape

Conv2D shape for tiling purposes (numeric dimensions).

MatMulShape

MatMul shape for tiling purposes (numeric dimensions).

TileConfig

Configuration for a single tile dimension.

TilingDefaults

Default tile sizes for common operations.

TilingPass

Tiling pass that classifies entry points and computes tiling plans.

TilingPlan

A tiling plan for a single operation.

Functions

tile_conv2d

Compute a tiling plan for a Conv2D operation.

tile_matmul

Compute a tiling plan for a MatMul operation.