tf.distribute.experimental.TPUStrategy