cubecl_reduce

Function reduce_kernel_virtual

Source 
pub fn reduce_kernel_virtual<In: Numeric, Out: Numeric, Acc: Numeric, R: ReduceFamily>(
    input: &VirtualTensor<In>,
    output: &mut VirtualTensor<Out, ReadWrite>,
    axis_reduce: u32,
    params: ReduceParams,
    config: R::Config,
)