Struct GlobalFullUnitReduce 

pub struct GlobalFullUnitReduce;

Implementations

impl GlobalFullUnitReduce

pub fn execute<P: ReducePrecision, Out: NumericVector, I: ReduceInstruction<P>>( input: &VirtualTensor<P::EI, P::SI>, output: &mut VirtualTensor<Out::T, Out::N, ReadWrite>, reduce_axis: usize, out_vec_axis: usize, inst: &I, vectorization_mode: VectorizationMode, blueprint: UnitReduceBlueprint, )

pub fn reduce_single<P: ReducePrecision, Out: NumericVector, I: ReduceInstruction<P>>( input: &VirtualTensor<P::EI, P::SI>, output: &mut VirtualTensor<Out::T, Out::N, ReadWrite>, reduce_axis: usize, reduce_index: usize, inst: &I, idle: ComptimeOption<bool>, vectorization_mode: VectorizationMode, ) -> Accumulator<P>

pub fn __expand_execute<P: ReducePrecision, Out: NumericVector, I: ReduceInstruction<P>>( scope: &mut Scope, input: <VirtualTensor<P::EI, P::SI> as RudaType>::ExpandType, output: <VirtualTensor<Out::T, Out::N, ReadWrite> as RudaType>::ExpandType, reduce_axis: <usize as RudaType>::ExpandType, out_vec_axis: <usize as RudaType>::ExpandType, inst: <I as RudaType>::ExpandType, vectorization_mode: VectorizationMode, blueprint: UnitReduceBlueprint, ) -> <() as RudaType>::ExpandType

pub fn __expand_reduce_single<P: ReducePrecision, Out: NumericVector, I: ReduceInstruction<P>>( scope: &mut Scope, input: <VirtualTensor<P::EI, P::SI> as RudaType>::ExpandType, output: <VirtualTensor<Out::T, Out::N, ReadWrite> as RudaType>::ExpandType, reduce_axis: <usize as RudaType>::ExpandType, reduce_index: <usize as RudaType>::ExpandType, inst: <I as RudaType>::ExpandType, idle: <ComptimeOption<bool> as RudaType>::ExpandType, vectorization_mode: VectorizationMode, ) -> <Accumulator<P> as RudaType>::ExpandType

Trait Implementations

impl RudaType for GlobalFullUnitReduce

type ExpandType = GlobalFullUnitReduceExpand