Struct Unloader

pub struct Unloader<EG: Numeric> {
    pub tensor_view: TensorWriter<EG>,
}

Fields

tensor_view: TensorWriter<EG>

Implementations

impl<EG: Numeric> Unloader<EG>

pub fn new( tensor: VirtualTensor<EG, ReadWrite>, x_offset: u32, y_offset: u32, batch_offset: u32, ) -> Self

pub fn __expand_new( context: &mut Scope, tensor: <VirtualTensor<EG, ReadWrite> as CubeType>::ExpandType, x_offset: <u32 as CubeType>::ExpandType, y_offset: <u32 as CubeType>::ExpandType, batch_offset: <u32 as CubeType>::ExpandType, ) -> <Self as CubeType>::ExpandType

Trait Implementations

impl<EG: Numeric> CubeType for Unloader<EG>

type ExpandType = UnloaderExpand<EG>

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

Wrapper around the init method, necessary to type inference.

impl<EG: Numeric> OutputLoader<EG> for Unloader<EG>

type StageWriter = Unloader<EG>

fn as_stage_writer<G: GlobalConfig>(this: Self) -> Self::StageWriter

fn __expand_as_stage_writer<G: GlobalConfig>( context: &mut Scope, this: <Self as CubeType>::ExpandType, ) -> <Self::StageWriter as CubeType>::ExpandType

impl<EG: Numeric> StageWriter<EG> for Unloader<EG>

fn write<ES: Numeric, G: GlobalConfig>( this: &mut Self, slice: Slice<Line<ES>>, tile_row: u32, tile_col: u32, config: G, )

Writes the given slice to global memory, at a position that depends on plane and accumulator indexes.

fn __expand_write<ES: Numeric, G: GlobalConfig>( context: &mut Scope, this: <Self as CubeType>::ExpandType, slice: <Slice<Line<ES>> as CubeType>::ExpandType, tile_row: <u32 as CubeType>::ExpandType, tile_col: <u32 as CubeType>::ExpandType, config: G, ) -> <() as CubeType>::ExpandType