Tensorflow distributed training on multiple nodes