Tensorflow Distributed Training Memory