Struct TensorInputExpand

pub struct TensorInputExpand<EI: Numeric, EO: Numeric, GA: MatmulArgs> { /* private fields */ }

Expand type for tensor input.

Implementations

impl<EI: Numeric, EO: Numeric, MA: MatmulArgs> TensorInputExpand<EI, EO, MA>

pub fn __expand_read_window_method( self, context: &mut Scope, start: <u32 as CubeType>::ExpandType, end: <u32 as CubeType>::ExpandType, ) -> <Slice<Line<EI>> as CubeType>::ExpandType

pub fn __expand_read_method( self, context: &mut Scope, coordinate: <u32 as CubeType>::ExpandType, ) -> <Line<EI> as CubeType>::ExpandType

pub fn __expand_shape_method( self, context: &mut Scope, axis: <u32 as CubeType>::ExpandType, ) -> <u32 as CubeType>::ExpandType

pub fn __expand_stride_method( self, context: &mut Scope, axis: <u32 as CubeType>::ExpandType, ) -> <u32 as CubeType>::ExpandType

pub fn __expand_rank_method( self, context: &mut Scope, ) -> <u32 as CubeType>::ExpandType

pub fn __expand_len_method( self, context: &mut Scope, ) -> <u32 as CubeType>::ExpandType

pub fn __expand_buffer_len_method( self, context: &mut Scope, ) -> <u32 as CubeType>::ExpandType

pub fn __expand_as_tensor_map_method( self, context: &mut Scope, ) -> <TensorMap<EI> as CubeType>::ExpandType

pub fn __expand_new( context: &mut Scope, state: <MA::State<EI, EO> as CubeType>::ExpandType, ident: TensorInputIdent, ) -> <TensorInput<EI, EO, MA> as CubeType>::ExpandType

pub fn __expand_read_window( context: &mut Scope, this: Self, start: <u32 as CubeType>::ExpandType, end: <u32 as CubeType>::ExpandType, ) -> <Slice<Line<EI>> as CubeType>::ExpandType

pub fn __expand_read( context: &mut Scope, this: Self, coordinate: <u32 as CubeType>::ExpandType, ) -> <Line<EI> as CubeType>::ExpandType

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

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

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

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

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

pub fn __expand_as_tensor_map( context: &mut Scope, this: Self, ) -> <TensorMap<EI> as CubeType>::ExpandType

Trait Implementations

impl<EI: Numeric, EO: Numeric, GA: MatmulArgs> Clone for TensorInputExpand<EI, EO, GA>

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, EO: Numeric, GA: MatmulArgs> CubeDebug for TensorInputExpand<EI, EO, GA>

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, EO: Numeric, GA: MatmulArgs> Init for TensorInputExpand<EI, EO, GA>

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

Initialize a type within a scope.

impl<EI: Numeric, EO: Numeric, MA: MatmulArgs> VirtualTensorOperationsExpand<EI> for TensorInputExpand<EI, EO, MA>

fn __expand_read_method( &self, scope: &mut Scope, index: ExpandElementTyped<u32>, ) -> ExpandElementTyped<Line<EI>>

fn __expand_read_window_method( &self, context: &mut Scope, start: ExpandElementTyped<u32>, end: ExpandElementTyped<u32>, ) -> ExpandElementTyped<Slice<Line<EI>>>

fn __expand_write_method( &self, _scope: &mut Scope, _index: ExpandElementTyped<u32>, _value: ExpandElementTyped<Line<EI>>, )

fn __expand_shape_method( &self, scope: &mut Scope, axis: ExpandElementTyped<u32>, ) -> ExpandElementTyped<u32>

fn __expand_stride_method( &self, scope: &mut Scope, axis: ExpandElementTyped<u32>, ) -> ExpandElementTyped<u32>

fn __expand_rank_method(&self, scope: &mut Scope) -> ExpandElementTyped<u32>

fn __expand_len_method(&self, scope: &mut Scope) -> ExpandElementTyped<u32>

fn __expand_buffer_len_method( &self, scope: &mut Scope, ) -> ExpandElementTyped<u32>

fn __expand_as_tensor_map_method( &self, scope: &mut Scope, ) -> ExpandElementTyped<TensorMap<EI>>