cubecl_convolution::loader::im2col_tma

Type Alias TmaIm2colReader

Source 
pub type TmaIm2colReader<MP> = FullStageToTileReader<<MP as MatmulPrecision>::ES, TmaIm2colTiling>;

Aliased Type§

pub struct TmaIm2colReader<MP> {
    pub stage_memory: StageMemory<<MP as MatmulPrecision>::ES, ContiguousTilingLayout<ColMajorTilingOrder>>,
    pub input_ident: InputIdent,
}

Fields§

§stage_memory: StageMemory<<MP as MatmulPrecision>::ES, ContiguousTilingLayout<ColMajorTilingOrder>>

Stage memory from which to read

§input_ident: InputIdent

Ident of the stage memory