cubecl_reduce::reduce_kernel

Function expand

Source 
pub fn expand<In: Numeric, Out: Numeric, Acc: Numeric, R: ReduceFamily, RA: ReduceArgs>(
    scope: &mut Scope,
    input: <RA::Input<In> as CubeType>::ExpandType,
    output: <RA::Output<Out> as CubeType>::ExpandType,
    axis_reduce: <u32 as CubeType>::ExpandType,
    params: ReduceParams,
    config: R::Config,
) -> <() as CubeType>::ExpandType