cubecl_reduce::reduce_kernel_virtual

Function expand

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