Skip to main content

convolution_matmul_selection

cubek_convolution::components

Function convolution_matmul_selection 

Source 
pub fn convolution_matmul_selection<R: Runtime>(
    tile_matmul: TileMatmulKind,
    client: &ComputeClient<R>,
    problem: &ConvolutionProblem,
    plane_dim: u32,
    swizzle: bool,
    vector_sizes: &MatmulVectorSizes,
    dtypes: &mut MatmulElems,
) -> Result<TilingBlueprint, MatmulAvailabilityError>