Skip to main content

ChainExpand

cubecl_std::tensor::layout::chain

Struct ChainExpand 

Source 
pub struct ChainExpand<L0: Layout, L1: Layout<SourceCoordinates = L0::Coordinates>> { /* private fields */ }

Trait Implementations§

Source§

impl<L0: Layout, L1: Layout<SourceCoordinates = L0::Coordinates>> AsMutExpand for ChainExpand<L0, L1>

Source§

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

Source§

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

Source§

impl<L0: Layout, L1: Layout<SourceCoordinates = L0::Coordinates>> AsRefExpand for ChainExpand<L0, L1>

Source§

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

Source§

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

Source§

impl<L0: Layout, L1: Layout<SourceCoordinates = L0::Coordinates>> CubeDebug for ChainExpand<L0, L1>

Source§

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
Source§

impl<L0: Layout, L1: Layout<SourceCoordinates = L0::Coordinates>> ExpandTypeClone for ChainExpand<L0, L1>

Source§

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.
Source§

impl<L0: Layout, L1: Layout<SourceCoordinates = L0::Coordinates>> IntoExpand for ChainExpand<L0, L1>

Source§

type Expand = ChainExpand<L0, L1>

Source§

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

Source§

impl<L0: Layout, L1: Layout<SourceCoordinates = L0::Coordinates>> IntoMut for ChainExpand<L0, L1>

Source§

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

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

impl<L0: Layout, L1: Layout<SourceCoordinates = L0::Coordinates>> LayoutExpand for ChainExpand<L0, L1>

Source§

type Coordinates = <L1 as Layout>::Coordinates

The coordinate type used by the conceptual tensor represented by this layout, i.e. (u32, u32, u32) for a fixed-rank 3D tensor. This does not have to match the rank of the underlying storage (if applicable). It’s only how the tensor is interpreted (viewed) by the code.
Source§

type SourceCoordinates = <L0 as Layout>::SourceCoordinates

The coordinate type used by the inner storage wrapped in this layout, i.e. u32 for Array, or (u32, u32) for a 2D view.
Source§

fn __expand_to_source_pos_method( &self, scope: &Scope, pos: <Self::Coordinates as CubeType>::ExpandType, ) -> <Self::SourceCoordinates as CubeType>::ExpandType

Source§

fn __expand_is_in_bounds_method( &self, scope: &Scope, pos: <Self::Coordinates as CubeType>::ExpandType, ) -> <bool as CubeType>::ExpandType

Source§

fn __expand_to_source_pos_checked_method( &self, scope: &Scope, pos: <Self::Coordinates as CubeType>::ExpandType, ) -> (<Self::SourceCoordinates as CubeType>::ExpandType, <bool as CubeType>::ExpandType)

Source§

fn __expand_shape_method( &self, scope: &Scope, ) -> <Self::Coordinates as CubeType>::ExpandType

Auto Trait Implementations§

§

impl<L0, L1> Freeze for ChainExpand<L0, L1>
where <L0 as CubeType>::ExpandType: Freeze, <L1 as CubeType>::ExpandType: Freeze,

§

impl<L0, L1> RefUnwindSafe for ChainExpand<L0, L1>
where <L0 as CubeType>::ExpandType: RefUnwindSafe, <L1 as CubeType>::ExpandType: RefUnwindSafe,

§

impl<L0, L1> Send for ChainExpand<L0, L1>
where <L0 as CubeType>::ExpandType: Send, <L1 as CubeType>::ExpandType: Send,

§

impl<L0, L1> Sync for ChainExpand<L0, L1>
where <L0 as CubeType>::ExpandType: Sync, <L1 as CubeType>::ExpandType: Sync,

§

impl<L0, L1> Unpin for ChainExpand<L0, L1>
where <L0 as CubeType>::ExpandType: Unpin, <L1 as CubeType>::ExpandType: Unpin,

§

impl<L0, L1> UnsafeUnpin for ChainExpand<L0, L1>
where <L0 as CubeType>::ExpandType: UnsafeUnpin, <L1 as CubeType>::ExpandType: UnsafeUnpin,

§

impl<L0, L1> UnwindSafe for ChainExpand<L0, L1>
where <L0 as CubeType>::ExpandType: UnwindSafe, <L1 as CubeType>::ExpandType: UnwindSafe,

Blanket Implementations§

Source§

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

Source§

fn type_id(&self) -> TypeId

Gets the TypeId of self. Read more
Source§

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

Source§

fn borrow(&self) -> &T

Immutably borrows from an owned value. Read more
Source§

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

Source§

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

Mutably borrows from an owned value. Read more
Source§

impl<ST, DT> CastableFrom<ST, Initialized, Initialized> for DT
where ST: ?Sized, DT: ?Sized,

Source§

impl<ST, DT> CastableFrom<ST, Uninit, Uninit> for DT
where ST: ?Sized, DT: ?Sized,

Source§

impl<T> From<T> for T

Source§

fn from(t: T) -> T

Returns the argument unchanged.

Source§

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

Source§

fn into(self) -> U

Calls U::from(self).

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

Source§

impl<T> IntoComptime for T

Source§

fn comptime(self) -> Self

Source§

impl<T> IntoEither for T

Source§

fn into_either(self, into_left: bool) -> Either<Self, Self>

Converts self into a Left variant of Either<Self, Self> if into_left is true. Converts self into a Right variant of Either<Self, Self> otherwise. Read more
Source§

fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
where F: FnOnce(&Self) -> bool,

Converts self into a Left variant of Either<Self, Self> if into_left(&self) returns true. Converts self into a Right variant of Either<Self, Self> otherwise. Read more
Source§

impl<T> Read<Exclusive, BecauseExclusive> for T
where T: ?Sized,

Source§

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

Source§

type Error = Infallible

The type returned in the event of a conversion error.
Source§

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

Performs the conversion.
Source§

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

Source§

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

The type returned in the event of a conversion error.
Source§

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

Performs the conversion.
Source§

impl<L> VirtualLayoutOperationsExpand<<L as LayoutExpand>::Coordinates, <L as LayoutExpand>::SourceCoordinates> for L
where L: LayoutExpand,

Source§

fn __expand_to_source_pos_virt_method( &self, scope: &Scope, pos: <<L as LayoutExpand>::Coordinates as CubeType>::ExpandType, ) -> <<L as LayoutExpand>::SourceCoordinates as CubeType>::ExpandType

Source§

fn __expand_to_source_pos_checked_virt_method( &self, scope: &Scope, pos: <<L as LayoutExpand>::Coordinates as CubeType>::ExpandType, ) -> <(<L as LayoutExpand>::SourceCoordinates, bool) as CubeType>::ExpandType

Source§

fn __expand_shape_virt_method( &self, scope: &Scope, ) -> <<L as LayoutExpand>::Coordinates as CubeType>::ExpandType

Source§

fn __expand_is_in_bounds_virt_method( &self, scope: &Scope, pos: <<L as LayoutExpand>::Coordinates as CubeType>::ExpandType, ) -> NativeExpand<bool>