Skip to main content

expand

ruprim::reduce::launch::reduce_kernel

Function expand 

Source 
pub fn expand<In: Numeric, InSize: Size, Out: Numeric, OutSize: Size, Acc: Numeric, RA: ReduceArgs>(
    scope: &mut Scope,
    _in_size: usize,
    _out_size: usize,
    input: <RA::Input<In, InSize> as RudaType>::ExpandType,
    output: <RA::Output<Out, OutSize> as RudaType>::ExpandType,
    reduce_axis: <usize as RudaType>::ExpandType,
    out_vec_axis: <usize as RudaType>::ExpandType,
    blueprint: ReduceBlueprint,
    config: ReduceOperationConfig,
    _input_dtype: StorageType,
    _output_dtype: StorageType,
    _acc_dtype: StorageType,
) -> <() as RudaType>::ExpandType