Struct cubecl::frontend::ExpandElementTyped

pub struct ExpandElementTyped<T>where
    T: CubeType,
{ /* private fields */ }

Expand description

Expand type associated with a type.

Implementations§

impl<C> ExpandElementTyped<Array<C>>
where C: CubePrimitive,

pub fn __expand_to_vectorized_method( self, context: &mut CubeContext, vectorization_factor: UInt, ) -> ExpandElementTyped<C>

impl<T> ExpandElementTyped<T>
where T: CubeType,

pub fn new(expand: ExpandElement) -> ExpandElementTyped<T>

Constructs a new ExpandElementTyped.

impl<T> ExpandElementTyped<T>
where T: CubePrimitive,

pub fn from_lit<L>(lit: L) -> ExpandElementTyped<T>
where L: Into<Variable>,

Create an ExpandElementTyped from a value that is normaly a literal.

pub fn constant(&self) -> Option<ConstantScalarValue>

Get the ConstantScalarValue from the variable.

impl<T> ExpandElementTyped<T>
where T: CubeType,

pub fn __expand_stride_method<C>( self, context: &mut CubeContext, dim: C, ) -> ExpandElementTyped<UInt>
where C: Index,

pub fn __expand_shape_method<C>( self, context: &mut CubeContext, dim: C, ) -> ExpandElementTyped<UInt>
where C: Index,

pub fn __expand_len_method( self, context: &mut CubeContext, ) -> ExpandElementTyped<UInt>

pub fn __expand_rank_method( self, _context: &mut CubeContext, ) -> ExpandElementTyped<UInt>

Trait Implementations§

impl<T> Clone for ExpandElementTyped<T>
where T: CubeType,

fn clone(&self) -> ExpandElementTyped<T>

Returns a copy of the value. Read more

1.0.0 · source§

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

Performs copy-assignment from source. Read more

impl<T> From<ExpandElement> for ExpandElementTyped<T>
where T: CubeType,

fn from(expand: ExpandElement) -> ExpandElementTyped<T>

Converts to this type from the input type.

impl<T> From<ExpandElementTyped<T>> for ExpandElement
where T: CubeType,

fn from(value: ExpandElementTyped<T>) -> ExpandElement

Converts to this type from the input type.

impl From<F32> for ExpandElementTyped<UInt>

fn from(value: F32) -> ExpandElementTyped<UInt>

Converts to this type from the input type.

impl From<F64> for ExpandElementTyped<UInt>

fn from(value: F64) -> ExpandElementTyped<UInt>

Converts to this type from the input type.

impl From<I32> for ExpandElementTyped<UInt>

fn from(value: I32) -> ExpandElementTyped<UInt>

Converts to this type from the input type.

impl From<I64> for ExpandElementTyped<UInt>

fn from(value: I64) -> ExpandElementTyped<UInt>

Converts to this type from the input type.

impl From<UInt> for ExpandElementTyped<UInt>

fn from(value: UInt) -> ExpandElementTyped<UInt>

Converts to this type from the input type.

impl From<bool> for ExpandElementTyped<bool>

fn from(value: bool) -> ExpandElementTyped<bool>

Converts to this type from the input type.

impl From<f32> for ExpandElementTyped<F32>

fn from(value: f32) -> ExpandElementTyped<F32>

Converts to this type from the input type.

impl From<f64> for ExpandElementTyped<F64>

fn from(value: f64) -> ExpandElementTyped<F64>

Converts to this type from the input type.

impl From<i32> for ExpandElementTyped<I32>

fn from(value: i32) -> ExpandElementTyped<I32>

Converts to this type from the input type.

impl From<i64> for ExpandElementTyped<I64>

fn from(value: i64) -> ExpandElementTyped<I64>

Converts to this type from the input type.

impl From<u32> for ExpandElementTyped<UInt>

fn from(value: u32) -> ExpandElementTyped<UInt>

Converts to this type from the input type.

impl<T> Init for ExpandElementTyped<SharedMemory<T>>
where T: CubePrimitive,

fn init(self, _context: &mut CubeContext) -> ExpandElementTyped<SharedMemory<T>>

Initialize a type within a context. Read more

impl<'a, C> Init for ExpandElementTyped<Slice<'a, C>>
where C: CubeType,

fn init(self, _context: &mut CubeContext) -> ExpandElementTyped<Slice<'a, C>>

Initialize a type within a context. Read more

impl<'a, C> Init for ExpandElementTyped<SliceMut<'a, C>>
where C: CubeType,

fn init(self, _context: &mut CubeContext) -> ExpandElementTyped<SliceMut<'a, C>>

Initialize a type within a context. Read more

impl<T> Init for ExpandElementTyped<T>
where T: ExpandElementBaseInit,

fn init(self, context: &mut CubeContext) -> ExpandElementTyped<T>

Initialize a type within a context. Read more

impl<E> SliceOperatorExpand<E> for ExpandElementTyped<Array<E>>
where E: CubePrimitive,

fn slice_base<Start, End>( &self, context: &mut CubeContext, start: Start, end: End, ) -> ExpandElement
where Start: Index, End: Index,

fn __expand_slice_method<Start, End>( &self, context: &mut CubeContext, start: Start, end: End, ) -> ExpandElementTyped<Slice<'static, E>>
where Start: Index, End: Index,

fn __expand_slice_mut_method<Start, End>( &self, context: &mut CubeContext, start: Start, end: End, ) -> ExpandElementTyped<SliceMut<'static, E>>
where Start: Index, End: Index,

fn __expand_slice_mut_unsafe_method<Start, End>( &self, context: &mut CubeContext, start: Start, end: End, ) -> ExpandElementTyped<SliceMut<'static, E>>
where Start: Index, End: Index,

fn __expand_as_slice_method( &self, _context: &mut CubeContext, ) -> ExpandElementTyped<Slice<'static, E>>

fn __expand_as_slice_mut_unsafe_method( &self, context: &mut CubeContext, ) -> ExpandElementTyped<SliceMut<'static, E>>

fn __expand_as_slice_mut_method( &self, _context: &mut CubeContext, ) -> ExpandElementTyped<SliceMut<'static, E>>

impl<E> SliceOperatorExpand<E> for ExpandElementTyped<SharedMemory<E>>
where E: CubePrimitive,

fn slice_base<Start, End>( &self, context: &mut CubeContext, start: Start, end: End, ) -> ExpandElement
where Start: Index, End: Index,

fn __expand_slice_method<Start, End>( &self, context: &mut CubeContext, start: Start, end: End, ) -> ExpandElementTyped<Slice<'static, E>>
where Start: Index, End: Index,

fn __expand_slice_mut_method<Start, End>( &self, context: &mut CubeContext, start: Start, end: End, ) -> ExpandElementTyped<SliceMut<'static, E>>
where Start: Index, End: Index,

fn __expand_slice_mut_unsafe_method<Start, End>( &self, context: &mut CubeContext, start: Start, end: End, ) -> ExpandElementTyped<SliceMut<'static, E>>
where Start: Index, End: Index,

fn __expand_as_slice_method( &self, _context: &mut CubeContext, ) -> ExpandElementTyped<Slice<'static, E>>

fn __expand_as_slice_mut_unsafe_method( &self, context: &mut CubeContext, ) -> ExpandElementTyped<SliceMut<'static, E>>

fn __expand_as_slice_mut_method( &self, _context: &mut CubeContext, ) -> ExpandElementTyped<SliceMut<'static, E>>

impl<'a, E> SliceOperatorExpand<E> for ExpandElementTyped<Slice<'a, E>>
where E: CubePrimitive,

fn slice_base<Start, End>( &self, context: &mut CubeContext, start: Start, end: End, ) -> ExpandElement
where Start: Index, End: Index,

fn __expand_slice_method<Start, End>( &self, context: &mut CubeContext, start: Start, end: End, ) -> ExpandElementTyped<Slice<'static, E>>
where Start: Index, End: Index,

fn __expand_slice_mut_method<Start, End>( &self, context: &mut CubeContext, start: Start, end: End, ) -> ExpandElementTyped<SliceMut<'static, E>>
where Start: Index, End: Index,

fn __expand_slice_mut_unsafe_method<Start, End>( &self, context: &mut CubeContext, start: Start, end: End, ) -> ExpandElementTyped<SliceMut<'static, E>>
where Start: Index, End: Index,

fn __expand_as_slice_method( &self, _context: &mut CubeContext, ) -> ExpandElementTyped<Slice<'static, E>>

fn __expand_as_slice_mut_unsafe_method( &self, context: &mut CubeContext, ) -> ExpandElementTyped<SliceMut<'static, E>>

fn __expand_as_slice_mut_method( &self, _context: &mut CubeContext, ) -> ExpandElementTyped<SliceMut<'static, E>>

impl<'a, E> SliceOperatorExpand<E> for ExpandElementTyped<SliceMut<'a, E>>
where E: CubePrimitive,

fn slice_base<Start, End>( &self, context: &mut CubeContext, start: Start, end: End, ) -> ExpandElement
where Start: Index, End: Index,

fn __expand_slice_method<Start, End>( &self, context: &mut CubeContext, start: Start, end: End, ) -> ExpandElementTyped<Slice<'static, E>>
where Start: Index, End: Index,

fn __expand_slice_mut_method<Start, End>( &self, context: &mut CubeContext, start: Start, end: End, ) -> ExpandElementTyped<SliceMut<'static, E>>
where Start: Index, End: Index,

fn __expand_slice_mut_unsafe_method<Start, End>( &self, context: &mut CubeContext, start: Start, end: End, ) -> ExpandElementTyped<SliceMut<'static, E>>
where Start: Index, End: Index,

fn __expand_as_slice_method( &self, _context: &mut CubeContext, ) -> ExpandElementTyped<Slice<'static, E>>

fn __expand_as_slice_mut_unsafe_method( &self, context: &mut CubeContext, ) -> ExpandElementTyped<SliceMut<'static, E>>

fn __expand_as_slice_mut_method( &self, _context: &mut CubeContext, ) -> ExpandElementTyped<SliceMut<'static, E>>

impl<E> SliceOperatorExpand<E> for ExpandElementTyped<Tensor<E>>
where E: CubePrimitive,

fn slice_base<Start, End>( &self, context: &mut CubeContext, start: Start, end: End, ) -> ExpandElement
where Start: Index, End: Index,

fn __expand_slice_method<Start, End>( &self, context: &mut CubeContext, start: Start, end: End, ) -> ExpandElementTyped<Slice<'static, E>>
where Start: Index, End: Index,

fn __expand_slice_mut_method<Start, End>( &self, context: &mut CubeContext, start: Start, end: End, ) -> ExpandElementTyped<SliceMut<'static, E>>
where Start: Index, End: Index,

fn __expand_slice_mut_unsafe_method<Start, End>( &self, context: &mut CubeContext, start: Start, end: End, ) -> ExpandElementTyped<SliceMut<'static, E>>
where Start: Index, End: Index,

fn __expand_as_slice_method( &self, _context: &mut CubeContext, ) -> ExpandElementTyped<Slice<'static, E>>

fn __expand_as_slice_mut_unsafe_method( &self, context: &mut CubeContext, ) -> ExpandElementTyped<SliceMut<'static, E>>

fn __expand_as_slice_mut_method( &self, _context: &mut CubeContext, ) -> ExpandElementTyped<SliceMut<'static, E>>

impl<T> Vectorized for ExpandElementTyped<T>
where T: CubeType,

fn vectorization_factor(&self) -> UInt

fn vectorize(self, factor: UInt) -> ExpandElementTyped<T>

Auto Trait Implementations§

impl<T> Freeze for ExpandElementTyped<T>

impl<T> RefUnwindSafe for ExpandElementTyped<T>
where T: RefUnwindSafe,

impl<T> !Send for ExpandElementTyped<T>

impl<T> !Sync for ExpandElementTyped<T>

impl<T> Unpin for ExpandElementTyped<T>
where T: Unpin,

impl<T> UnwindSafe for ExpandElementTyped<T>
where T: UnwindSafe,

Blanket Implementations§

impl<T> Any for T
where T: 'static + ?Sized,

fn type_id(&self) -> TypeId

Gets the TypeId of self. Read more

impl<T> Borrow<T> for T
where T: ?Sized,

fn borrow(&self) -> &T

Immutably borrows from an owned value. Read more

impl<T> BorrowMut<T> for T
where T: ?Sized,

fn borrow_mut(&mut self) -> &mut T

Mutably borrows from an owned value. Read more

impl<T> CloneToUninit for T
where T: Clone,

default unsafe fn clone_to_uninit(&self, dst: *mut T)

🔬This is a nightly-only experimental API. (clone_to_uninit)

Performs copy-assignment from self to dst. Read more

impl<T> From<T> for T

fn from(t: T) -> T

Returns the argument unchanged.

impl<T, U> Into<U> for T
where U: From<T>,

fn into(self) -> U

Calls U::from(self).

That is, this conversion is whatever the implementation of From<T> for U chooses to do.

impl<T> ToOwned for T
where T: Clone,

type Owned = T

The resulting type after obtaining ownership.

fn to_owned(&self) -> T

Creates owned data from borrowed data, usually by cloning. Read more

fn clone_into(&self, target: &mut T)

Uses borrowed data to replace owned data, usually by cloning. Read more

impl<T, U> TryFrom<U> for T
where U: Into<T>,

type Error = Infallible

The type returned in the event of a conversion error.

fn try_from(value: U) -> Result<T, <T as TryFrom<U>>::Error>

Performs the conversion.

impl<T, U> TryInto<U> for T
where U: TryFrom<T>,

type Error = <U as TryFrom<T>>::Error

The type returned in the event of a conversion error.

fn try_into(self) -> Result<U, <U as TryFrom<T>>::Error>

Performs the conversion.

impl<V, T> VZip<V> for T
where V: MultiLane<T>,

fn vzip(self) -> V