Как использовать TPU в пользовательском цикле обучения TensorFlow?

Question

Я не использую Keras. И я хочу использовать TPU в Google Colab. Вопросы:

1. Может ли tf.session автоматически использовать TPU?
2. Что делают tf.contrib.tpu.TPUDistributionStrategy, tf.contrib.tpu.rewrite, tf.contrib.cluster_resolver.TPUClusterResolver в вычислениях TPU? Они все необходимы?