tf.distribute.experimental.CommunicationImplementation