Skip to main content

reduce_kernel

ruprim::reduce::launch

Function reduce_kernel 

Source 
pub fn reduce_kernel<In: Numeric, InSize: Size, Out: Numeric, OutSize: Size, Acc: Numeric, RA: ReduceArgs>(
    input: &RA::Input<In, InSize>,
    output: &mut RA::Output<Out, OutSize>,
    reduce_axis: usize,
    out_vec_axis: usize,
    blueprint: ReduceBlueprint,
    config: ReduceOperationConfig,
    _input_dtype: StorageType,
    _output_dtype: StorageType,
    _acc_dtype: StorageType,
)