Gradient for reduce_sum