Function split_reduceop 

pub fn split_reduceop(
    reduce: &Arc<UOp>,
    config: &SplitReduceOpConfig,
) -> Option<Arc<UOp>>

Split large REDUCE_AXIS into two stages.