Struct ArgAccumulator

pub struct ArgAccumulator<N: Numeric> {
    pub elements: SharedMemory<Line<N>>,
    pub args: SharedMemory<Line<u32>>,
}

A pair of shared memory used for ArgMax and ArgMin.

Fields

elements: SharedMemory<Line<N>>

args: SharedMemory<Line<u32>>

Trait Implementations

impl<N: Numeric> CubeType for ArgAccumulator<N>

type ExpandType = ArgAccumulatorExpand<N>

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

Wrapper around the init method, necessary to type inference.

impl<In: Numeric> SharedAccumulator for ArgAccumulator<In>

type Item = (Line<In>, Line<u32>)

fn allocate(length: u32, line_size: u32, _coordinate: bool) -> Self

fn read(accumulator: &Self, index: u32) -> Self::Item

fn write(accumulator: &mut Self, index: u32, item: Self::Item)

fn __expand_allocate( scope: &mut Scope, length: u32, line_size: u32, _coordinate: bool, ) -> <Self as CubeType>::ExpandType

fn __expand_read( scope: &mut Scope, accumulator: <Self as CubeType>::ExpandType, index: <u32 as CubeType>::ExpandType, ) -> <Self::Item as CubeType>::ExpandType

fn __expand_write( scope: &mut Scope, accumulator: <Self as CubeType>::ExpandType, index: <u32 as CubeType>::ExpandType, item: <Self::Item as CubeType>::ExpandType, ) -> <() as CubeType>::ExpandType