Make sure the large sum in the duality gap computation happens on the PS

and we don't send all values to the worker.
Replace special values_reduce_sum() method with a generic export_sharded().
Other minor cleanups.
Change: 131424317