Struct TensorReaderExpand

pub struct TensorReaderExpand<EI: Numeric> {
    pub tensor: <VirtualTensor<EI> as CubeType>::ExpandType,
    pub x_offset: <u32 as CubeType>::ExpandType,
    pub y_offset: <u32 as CubeType>::ExpandType,
    pub stride_x: <u32 as CubeType>::ExpandType,
    pub stride_y: <u32 as CubeType>::ExpandType,
    pub shape_x: <u32 as CubeType>::ExpandType,
    pub shape_y: <u32 as CubeType>::ExpandType,
    pub batch_offset: <u32 as CubeType>::ExpandType,
}

Fields

tensor: <VirtualTensor<EI> as CubeType>::ExpandType

x_offset: <u32 as CubeType>::ExpandType

y_offset: <u32 as CubeType>::ExpandType

stride_x: <u32 as CubeType>::ExpandType

stride_y: <u32 as CubeType>::ExpandType

shape_x: <u32 as CubeType>::ExpandType

shape_y: <u32 as CubeType>::ExpandType

batch_offset: <u32 as CubeType>::ExpandType

Implementations

impl<EG: Numeric> TensorReaderExpand<EG>

pub fn __expand_update_view_method( self, context: &mut Scope, k_offset: <u32 as CubeType>::ExpandType, ident: InputIdent, ) -> <() as CubeType>::ExpandType

pub fn __expand_load_window_in_tile_method<G: GlobalConfig>( self, context: &mut Scope, tile: <(u32, u32) as CubeType>::ExpandType, nth_window: <u32 as CubeType>::ExpandType, input_ident: InputIdent, config: G, ) -> <Window<EG> as CubeType>::ExpandType

pub fn __expand_load_window_in_stage_method<G: GlobalConfig>( self, context: &mut Scope, nth_window: <u32 as CubeType>::ExpandType, input_ident: InputIdent, config: G, ) -> <Window<EG> as CubeType>::ExpandType

pub fn __expand_load_coalesced_in_tile_method<G: GlobalConfig>( self, context: &mut Scope, tile_x: <u32 as CubeType>::ExpandType, tile_y: <u32 as CubeType>::ExpandType, position: <u32 as CubeType>::ExpandType, input_ident: InputIdent, config: G, ) -> <Line<EG> as CubeType>::ExpandType

pub fn __expand_load_coalesced_in_stage_method<G: GlobalConfig>( self, context: &mut Scope, position: <u32 as CubeType>::ExpandType, input_ident: InputIdent, config: G, ) -> <Line<EG> as CubeType>::ExpandType

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

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

pub fn __expand_load_window_in_tile<G: GlobalConfig>( context: &mut Scope, this: Self, tile: <(u32, u32) as CubeType>::ExpandType, nth_window: <u32 as CubeType>::ExpandType, input_ident: InputIdent, config: G, ) -> <Window<EG> as CubeType>::ExpandType

pub fn __expand_load_window_in_stage<G: GlobalConfig>( context: &mut Scope, this: Self, nth_window: <u32 as CubeType>::ExpandType, input_ident: InputIdent, config: G, ) -> <Window<EG> as CubeType>::ExpandType

pub fn __expand_load_coalesced_in_tile<G: GlobalConfig>( context: &mut Scope, this: Self, tile_x: <u32 as CubeType>::ExpandType, tile_y: <u32 as CubeType>::ExpandType, position: <u32 as CubeType>::ExpandType, input_ident: InputIdent, config: G, ) -> <Line<EG> as CubeType>::ExpandType

pub fn __expand_load_coalesced_in_stage<G: GlobalConfig>( context: &mut Scope, this: Self, position: <u32 as CubeType>::ExpandType, input_ident: InputIdent, config: G, ) -> <Line<EG> as CubeType>::ExpandType

Trait Implementations

impl<EI: Numeric> Clone for TensorReaderExpand<EI>

fn clone(&self) -> Self

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl<EI: Numeric> CubeDebug for TensorReaderExpand<EI>

fn set_debug_name(&self, scope: &mut Scope, name: &'static str)

Set the debug name of this type’s expansion. Should do nothing for types that don’t appear at runtime

impl<EI: Numeric> Init for TensorReaderExpand<EI>

fn init(self, context: &mut Scope) -> Self

Initialize a type within a scope.