TensorFlow 2 with Horovod