pass in part_ops to AddReduce, so that it is syncd alongside the other data