Enum ScalesLayoutExpand 

pub enum ScalesLayoutExpand {
    PerTensor(<PerTensorLayout as CubeType>::ExpandType),
    BlockScaled(<BlockScaledLayout as CubeType>::ExpandType),
}

Variants

PerTensor(<PerTensorLayout as CubeType>::ExpandType)

BlockScaled(<BlockScaledLayout as CubeType>::ExpandType)

Implementations

impl ScalesLayoutExpand

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

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

Trait Implementations

impl Clone for ScalesLayoutExpand

fn clone(&self) -> Self

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl CubeDebug for ScalesLayoutExpand

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 IntoMut for ScalesLayoutExpand

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

impl LayoutExpand for ScalesLayoutExpand

type Coordinates = u32

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.

type SourceCoordinates = u32

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

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

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

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

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