Struct VirtualTensorExpand

pub struct VirtualTensorExpand<E: Numeric, IO> { /* private fields */ }

Expand type for VirtualTensor.

Implementations

impl<E: Numeric, IO: Clone> VirtualTensorExpand<E, IO>

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

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

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

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

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

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

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

pub fn __expand_read( scope: &mut Scope, this: Self, index: <u32 as CubeType>::ExpandType, ) -> <Line<E> as CubeType>::ExpandType

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

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

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

impl<E: Numeric, IO: Clone> VirtualTensorExpand<E, IO>

pub fn __expand_coordinate_method( self, scope: &mut Scope, index: <u32 as CubeType>::ExpandType, dim: <u32 as CubeType>::ExpandType, ) -> <u32 as CubeType>::ExpandType

pub fn __expand_coordinate( scope: &mut Scope, this: Self, index: <u32 as CubeType>::ExpandType, dim: <u32 as CubeType>::ExpandType, ) -> <u32 as CubeType>::ExpandType

Trait Implementations

impl<E: Clone + Numeric, IO: Clone> Clone for VirtualTensorExpand<E, IO>

fn clone(&self) -> VirtualTensorExpand<E, IO>

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl<E: Numeric, IO> CubeDebug for VirtualTensorExpand<E, IO>

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<E: Numeric, IO> IntoMut for VirtualTensorExpand<E, IO>

fn into_mut(self, _scope: &mut Scope) -> Self

impl<E: Numeric, IO: Clone> ListExpand<Line<E>> for VirtualTensorExpand<E, IO>

fn __expand_read_method( &self, scope: &mut Scope, index: <u32 as CubeType>::ExpandType, ) -> <Line<E> as CubeType>::ExpandType

fn __expand_read_unchecked_method( &self, _scope: &mut Scope, _index: ExpandElementTyped<u32>, ) -> <Line<E> as CubeType>::ExpandType

impl<E: Numeric> ListMutExpand<Line<E>> for VirtualTensorExpand<E, ReadWrite>

fn __expand_write_method( &self, scope: &mut Scope, index: <u32 as CubeType>::ExpandType, value: <Line<E> as CubeType>::ExpandType, ) -> <() as CubeType>::ExpandType