message: Update example for CTL in TF 1.x and disable partial batch support
diff: @@ -89,6 +89,9 @@ def main(argv):
  tpu=FLAGS.tpu)
  strategy = tf.contrib.distribute.TPUStrategy(cluster_resolver)
  + # TODO(rxsang): This doesn't work on cloud for some reason
  + strategy.extended.experimental_enable_get_next_as_optional = False
  +
  with strategy.scope():
  train_ds, test_ds = mnist_datasets()
  train_ds = train_ds.shuffle(NUM_TRAIN_IMAGES).batch(FLAGS.batch_size)
  @@ -151,7 +154,11 @@ def main(argv):
  test_loss.variables +
  test_accuracy.variables)
  - with tf.Session(cluster_resolver.master()) as session:
  + config = tf.ConfigProto()
  + config.cluster_def.CopyFrom(
  + cluster_resolver.cluster_spec().as_cluster_def())
  +
  + with tf.Session(cluster_resolver.master(), config=config) as session:
  session.run([v.initializer for v in all_variables])
  for epoch in range(0, FLAGS.num_epochs):
  