Struct VirtualTensorExpand 

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

Expand type for VirtualTensor.

Implementations

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

pub fn __expand_read_method( &self, scope: &Scope, index: <usize as CubeType>::ExpandType, ) -> <Vector<E, N> as CubeType>::ExpandType

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

pub fn __expand_read( scope: &Scope, this: &Self, index: <usize as CubeType>::ExpandType, ) -> <Vector<E, N> as CubeType>::ExpandType

pub fn __expand_len( scope: &Scope, this: &Self, ) -> <usize as CubeType>::ExpandType

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

pub fn __expand_as_slice_method<'infer, 'scope>( &'infer self, scope: &'scope Scope, ) -> &'infer <[Vector<E, N>] as CubeType>::ExpandType

pub fn __expand_as_mut_slice_method<'infer, 'scope>( &'infer mut self, scope: &'scope Scope, ) -> &'infer mut <[Vector<E, N>] as CubeType>::ExpandType

pub fn __expand_as_tensor_map_method( &self, scope: &Scope, ) -> <ComptimeOption<TensorMap<E, Tiled>> as CubeType>::ExpandType

pub fn __expand_slice_method<'infer, 'scope>( &'infer self, scope: &'scope Scope, start: <usize as CubeType>::ExpandType, end: <usize as CubeType>::ExpandType, ) -> &'infer <[Vector<E, N>] as CubeType>::ExpandType

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

Get the shape of the tensor at the given axis.

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

Get the stride of the tensor at the given axis.

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

Get the rank of the tensor.

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

pub fn __expand_as_slice<'infer, 'scope>( scope: &'scope Scope, this: &'infer Self, ) -> &'infer <[Vector<E, N>] as CubeType>::ExpandType

pub fn __expand_as_mut_slice<'infer, 'scope>( scope: &'scope Scope, this: &'infer mut Self, ) -> &'infer mut <[Vector<E, N>] as CubeType>::ExpandType

pub fn __expand_as_tensor_map( scope: &Scope, this: &Self, ) -> <ComptimeOption<TensorMap<E, Tiled>> as CubeType>::ExpandType

pub fn __expand_slice<'infer, 'scope>( scope: &'scope Scope, this: &'infer Self, start: <usize as CubeType>::ExpandType, end: <usize as CubeType>::ExpandType, ) -> &'infer <[Vector<E, N>] as CubeType>::ExpandType

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

Get the shape of the tensor at the given axis.

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

Get the stride of the tensor at the given axis.

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

Get the rank of the tensor.

pub fn __expand_buffer_len( scope: &Scope, this: &Self, ) -> <usize as CubeType>::ExpandType

impl<E: Numeric, N: Size, IO: Clone + 'static> VirtualTensorExpand<E, N, IO>

pub fn __expand_as_view_method<'infer, 'scope>( &'infer self, scope: &'scope Scope, ) -> <View<'infer, Vector<E, N>, usize> as CubeType>::ExpandType

Create a conceptual view over this tensor, with a simple linear layout

pub fn __expand_as_view<'infer, 'scope>( scope: &'scope Scope, this: &'infer Self, ) -> <View<'infer, Vector<E, N>, usize> as CubeType>::ExpandType

Create a conceptual view over this tensor, with a simple linear layout

impl<E: Numeric, N: Size, IO: Clone + 'static> VirtualTensorExpand<E, N, IO>

pub fn __expand_view_method<C: Coordinates + 'static>( &self, scope: &Scope, layout: VirtualLayoutExpand<C, Coords1d>, ) -> ViewExpand<'_, Vector<E, N>, C>

Create a conceptual view over this tensor, allowing for multi-dimensional indexing with custom layouts

pub fn __expand_into_view_method<C: Coordinates + 'static>( self, scope: &Scope, layout: VirtualLayoutExpand<C, Coords1d>, ) -> ViewExpand<'static, Vector<E, N>, C>

impl<E: Numeric, N: Size> VirtualTensorExpand<E, N, ReadWrite>

pub fn __expand_view_mut_method<C: Coordinates + 'static>( &mut self, scope: &Scope, layout: VirtualLayoutExpand<C, Coords1d>, ) -> ViewMutExpand<'_, Vector<E, N>, C>

pub fn __expand_into_view_mut_method<C: Coordinates + 'static>( self, scope: &Scope, layout: VirtualLayoutExpand<C, Coords1d>, ) -> ViewMutExpand<'static, Vector<E, N>, C>

impl<E: Numeric, N: Size> VirtualTensorExpand<E, N, ReadWrite>

pub fn __expand_as_view_mut_method<'infer, 'scope>( &'infer mut self, scope: &'scope Scope, ) -> <ViewMut<'infer, Vector<E, N>, usize> as CubeType>::ExpandType

Create a conceptual mutable view over this tensor, with a simple linear layout

pub fn __expand_as_view_mut<'infer, 'scope>( scope: &'scope Scope, this: &'infer mut Self, ) -> <ViewMut<'infer, Vector<E, N>, usize> as CubeType>::ExpandType

Create a conceptual mutable view over this tensor, with a simple linear layout

impl<E: Numeric, N: Size, IO: Clone + 'static> VirtualTensorExpand<E, N, IO>

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

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

impl<E: Numeric, N: Size> VirtualTensorExpand<E, N, ReadWrite>

pub fn __expand_write_method( &mut self, scope: &Scope, index: <usize as CubeType>::ExpandType, value: <Vector<E, N> as CubeType>::ExpandType, )

pub fn __expand_write( scope: &Scope, this: &mut Self, index: <usize as CubeType>::ExpandType, value: <Vector<E, N> as CubeType>::ExpandType, )

Trait Implementations

impl<E: Numeric, N: Size, IO: Clone> AsMutExpand for VirtualTensorExpand<E, N, IO>

fn __expand_ref_mut_method(&mut self, _: &Scope) -> &mut Self

fn __expand_as_mut_method(&mut self, scope: &Scope) -> &mut T

impl<E: Numeric, N: Size, IO: Clone> AsRefExpand for VirtualTensorExpand<E, N, IO>

fn __expand_ref_method(&self, _: &Scope) -> &Self

fn __expand_as_ref_method(&self, scope: &Scope) -> &T

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

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

Returns a duplicate of the value.

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

Performs copy-assignment from source.

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

fn set_debug_name(&self, scope: &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, N: Size, IO: Clone> ExpandTypeClone for VirtualTensorExpand<E, N, IO>

fn clone_unchecked(&self) -> Self

Unchecked clone that only clones the conceptual runtime value. Should only be used in cases where each copy is used in a mutually exclusive branch (i.e. match, runtime enums). This is intentionally separated from Rust’s Clone semantics and should only be used for the conceptual expand values, never real data. Using two values in the same branch is undefined behaviour.

impl<E: Numeric, N: Size, IO: Clone> IntoExpand for VirtualTensorExpand<E, N, IO>

type Expand = VirtualTensorExpand<E, N, IO>

fn into_expand(self, _: &Scope) -> Self::Expand

impl<E: Numeric, N: Size, IO> IntoMut for VirtualTensorExpand<E, N, IO>

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

Convert the variable into a potentially new mutable variable in scope, copying if needed.

impl<E: Numeric, N: Size, IO: Clone> SliceOperatorExpand<Vector<E, N>> for VirtualTensorExpand<E, N, IO>

fn __expand_slice_method( &self, scope: &Scope, start: NativeExpand<usize>, end: NativeExpand<usize>, ) -> &SliceExpand<Vector<E, N>>

fn __expand_slice_mut_method( &mut self, _: &Scope, _: NativeExpand<usize>, _: NativeExpand<usize>, ) -> &mut SliceExpand<Vector<E, N>>

impl<E: Numeric, N: Size, IO: Clone> VectorizedExpand for VirtualTensorExpand<E, N, IO>

fn vector_size(&self) -> VectorSize

fn __expand_vector_size_method(&self, _scope: &Scope) -> usize

impl<T: Numeric, N: Size, IO: Clone> ViewOperationsExpand<Vector<T, N>, usize> for VirtualTensorExpand<T, N, IO>

fn __expand_read_method( &self, scope: &Scope, pos: NativeExpand<usize>, ) -> <Vector<T, N> as CubeType>::ExpandType

fn __expand_read_checked_method( &self, scope: &Scope, pos: NativeExpand<usize>, ) -> <Vector<T, N> as CubeType>::ExpandType

fn __expand_read_masked_method( &self, scope: &Scope, pos: NativeExpand<usize>, mask_value: <Vector<T, N> as CubeType>::ExpandType, ) -> <Vector<T, N> as CubeType>::ExpandType

fn __expand_read_unchecked_method( &self, scope: &Scope, pos: NativeExpand<usize>, ) -> <Vector<T, N> as CubeType>::ExpandType

fn __expand_as_linear_slice_method( &self, scope: &Scope, pos: NativeExpand<usize>, end: NativeExpand<usize>, ) -> &SliceExpand<Vector<T, N>>

fn __expand_shape_method(&self, scope: &Scope) -> NativeExpand<usize>

fn __expand_is_in_bounds_method( &self, scope: &Scope, pos: NativeExpand<usize>, ) -> NativeExpand<bool>

fn __expand_tensor_map_load_method( &self, _scope: &Scope, _barrier: &NativeExpand<Barrier>, _shared_memory: &mut SliceExpand<Vector<T, N>>, _pos: NativeExpand<usize>, )

impl<T: Numeric, N: Size> ViewOperationsMutExpand<Vector<T, N>, usize> for VirtualTensorExpand<T, N, ReadWrite>

fn __expand_write_method( &self, scope: &Scope, pos: NativeExpand<usize>, value: <Vector<T, N> as CubeType>::ExpandType, )

fn __expand_write_checked_method( &self, scope: &Scope, pos: NativeExpand<usize>, value: <Vector<T, N> as CubeType>::ExpandType, )

fn __expand_as_linear_slice_mut_method( &self, _scope: &Scope, _pos: NativeExpand<usize>, _end: NativeExpand<usize>, ) -> &mut SliceExpand<Vector<T, N>>

fn __expand_tensor_map_store_method( &self, _scope: &Scope, _shared_memory: &SliceExpand<Vector<T, N>>, _pos: <Coords1d as CubeType>::ExpandType, )