Skip to main content

convolution_matmul_selection

cubek_convolution::components

Function convolution_matmul_selection 

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