Struct TensorWriterExpand 

pub struct TensorWriterExpand<EO: Numeric> {
    pub tensor: <VirtualTensor<EO, ReadWrite> 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<EO, ReadWrite> 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> TensorWriterExpand<EG>

pub fn __expand_write_coalesced_method<ES: Numeric, G: GlobalConfig>( self, context: &mut Scope, tile_x: <u32 as CubeType>::ExpandType, tile_y: <u32 as CubeType>::ExpandType, unit_id: <u32 as CubeType>::ExpandType, value: <Line<ES> as CubeType>::ExpandType, config: G, ) -> <() as CubeType>::ExpandType

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

pub fn __expand_write_coalesced<ES: Numeric, G: GlobalConfig>( context: &mut Scope, this: Self, tile_x: <u32 as CubeType>::ExpandType, tile_y: <u32 as CubeType>::ExpandType, unit_id: <u32 as CubeType>::ExpandType, value: <Line<ES> as CubeType>::ExpandType, config: G, ) -> <() as CubeType>::ExpandType

Trait Implementations

impl<EO: Numeric> Clone for TensorWriterExpand<EO>

fn clone(&self) -> Self

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl<EO: Numeric> CubeDebug for TensorWriterExpand<EO>

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<EO: Numeric> Init for TensorWriterExpand<EO>

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

Initialize a type within a scope.