Struct ArgAccumulator 

pub struct ArgAccumulator<P: ReducePrecision> {
    pub elements: SharedMemory<Vector<P::EA, P::SI>>,
    pub args: SharedMemory<Vector<u32, P::SI>>,
}

A pair of shared memory used for ArgMax and ArgMin.

Fields

elements: SharedMemory<Vector<P::EA, P::SI>>

args: SharedMemory<Vector<u32, P::SI>>

Trait Implementations

impl<P: ReducePrecision> RudaType for ArgAccumulator<P>

type ExpandType = ArgAccumulatorExpand<P>

impl<P: ReducePrecision, I: ReduceInstruction<P>> SharedAccumulator<P, I> for ArgAccumulator<P>

fn allocate(length: usize, _coordinate: bool, _inst: &I) -> Self

fn read(accumulator: &Self, index: usize) -> Accumulator<P>

fn write(accumulator: &mut Self, index: usize, item: Accumulator<P>)

fn __expand_allocate( scope: &mut Scope, length: usize, _coordinate: bool, _inst: <I as RudaType>::ExpandType, ) -> <Self as RudaType>::ExpandType

fn __expand_read( scope: &mut Scope, accumulator: <Self as RudaType>::ExpandType, index: <usize as RudaType>::ExpandType, ) -> <Accumulator<P> as RudaType>::ExpandType

fn __expand_write( scope: &mut Scope, accumulator: <Self as RudaType>::ExpandType, index: <usize as RudaType>::ExpandType, item: <Accumulator<P> as RudaType>::ExpandType, ) -> <() as RudaType>::ExpandType