Struct ErasedTensorExpand 

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

Expand type for ErasedTensor.

Implementations

impl<E: Numeric, IO> ErasedTensorExpand<E, IO>

pub fn new<S: ErasedBacking<E, IO> + 'static>(backing: S) -> Self

Erase backing’s width.

Takes the expand, because that is where a width is known: the caller has the N-typed backing in hand and gives up naming it here.

impl<E: Numeric, IO: ErasedIoRead> ErasedTensorExpand<E, IO>

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

Read one N-wide line at index, counted in lines.

Panics

At expansion, when N is not the width the tensor was built for.

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

Read one N-wide line at index, counted in lines.

Panics

At expansion, when N is not the width the tensor was built for.

impl<E: Numeric, IO: ErasedIoWrite> ErasedTensorExpand<E, IO>

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

Write one N-wide line at index, counted in lines.

Panics

At expansion, when N is not the width the tensor was built for.

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

Write one N-wide line at index, counted in lines.

Panics

At expansion, when N is not the width the tensor was built for.

impl<E: Numeric, IO: ErasedIo> ErasedTensorExpand<E, IO>

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

How many lines the tensor holds.

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

How many lines the tensor holds.

Trait Implementations

impl<E: Numeric, IO> AsMutExpand for ErasedTensorExpand<E, 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, IO> AsRefExpand for ErasedTensorExpand<E, IO>

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

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

impl<E: Numeric, IO> Clone for ErasedTensorExpand<E, IO>

fn clone(&self) -> Self

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl<E: Numeric, IO> CubeDebug for ErasedTensorExpand<E, 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, IO> ExpandTypeClone for ErasedTensorExpand<E, 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> From<VirtualTensorExpand<E, N, ReadOnly>> for ErasedTensorExpand<E, ReadOnly>

fn from(tensor: VirtualTensorExpand<E, N, ReadOnly>) -> Self

Converts to this type from the input type.

impl<E: Numeric, N: Size> From<VirtualTensorExpand<E, N, ReadWrite>> for ErasedTensorExpand<E, ReadWrite>

fn from(tensor: VirtualTensorExpand<E, N, ReadWrite>) -> Self

Converts to this type from the input type.

impl<E: Numeric, IO> IntoExpand for ErasedTensorExpand<E, IO>

type Expand = ErasedTensorExpand<E, IO>

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

impl<E: Numeric, IO> IntoMut for ErasedTensorExpand<E, 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, IO: ErasedIo> VectorizedExpand for ErasedTensorExpand<E, IO>

fn __expand_vector_size_method(&self, scope: &Scope) -> VectorSize

impl<E: Numeric, N: Size, IO: ErasedIo> ViewOperationsExpand<Vector<E, N>, usize> for ErasedTensorExpand<E, IO>

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

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

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

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

fn __expand_as_linear_slice_method( &self, _scope: &Scope, _pos: NativeExpand<usize>, _end: NativeExpand<usize>, ) -> &SliceExpand<Vector<E, 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<E, N>>, _pos: NativeExpand<usize>, )

impl<E: Numeric, N: Size, IO: ErasedIoWrite> ViewOperationsMutExpand<Vector<E, N>, usize> for ErasedTensorExpand<E, IO>

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

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

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

fn __expand_tensor_map_store_method( &self, _scope: &Scope, _shared_memory: &SliceExpand<Vector<E, N>>, _pos: NativeExpand<usize>, )