Struct SimpleIm2colLoader

pub struct SimpleIm2colLoader<MP: MatmulPrecision, G: ConvGemmConfig> {
    pub tensor_view: Im2colReader<MP::EI>,
    pub stage: Stage<MP::ES, ContiguousTilingLayout<RowMajorTilingOrder>>,
    /* private fields */
}

Loader that translates matrix coordinates to input coordinates using the im2col algorithm

Fields

tensor_view: Im2colReader<MP::EI>

stage: Stage<MP::ES, ContiguousTilingLayout<RowMajorTilingOrder>>

Implementations

impl<MP: MatmulPrecision, G: ConvGemmConfig> SimpleIm2colLoader<MP, G>

pub fn new( tensor: VirtualTensor<MP::EI>, x_offset: u32, y_offset: u32, runtime_args: &RuntimeArgs, config: G, ) -> Self

pub fn advance_view(this: &mut Self, k_offset: u32)

pub fn reader( this: &Self, ) -> FullReader<MP::ES, ContiguousTilingLayout<RowMajorTilingOrder>>

pub fn fill_stage(this: &mut Self, config: G)

pub fn __expand_new( context: &mut Scope, tensor: <VirtualTensor<MP::EI> as CubeType>::ExpandType, x_offset: <u32 as CubeType>::ExpandType, y_offset: <u32 as CubeType>::ExpandType, runtime_args: <RuntimeArgs as CubeType>::ExpandType, config: G, ) -> <Self as CubeType>::ExpandType

pub fn __expand_advance_view( context: &mut Scope, this: <Self as CubeType>::ExpandType, k_offset: <u32 as CubeType>::ExpandType, ) -> <() as CubeType>::ExpandType

pub fn __expand_reader( context: &mut Scope, this: <Self as CubeType>::ExpandType, ) -> <FullReader<MP::ES, ContiguousTilingLayout<RowMajorTilingOrder>> as CubeType>::ExpandType

pub fn __expand_fill_stage( context: &mut Scope, this: <Self as CubeType>::ExpandType, config: G, ) -> <() as CubeType>::ExpandType

Trait Implementations

impl<MP: MatmulPrecision, G: ConvGemmConfig> CubeType for SimpleIm2colLoader<MP, G>

type ExpandType = SimpleIm2colLoaderExpand<MP, G>

fn init(scope: &mut Scope, expand: Self::ExpandType) -> Self::ExpandType

Wrapper around the init method, necessary to type inference.