Merge pull request #3336 from tensorflow/frankchn-patch-1

Use the short path for TPUClusterResolver

Merge pull request #3336 from tensorflow/frankchn-patch-1
Use the short path for TPUClusterResolver
b7edd6e3 · Neal Wu · GitHub · aff0c314 · 9e23bab8 · b7edd6e3
Unverified Commit b7edd6e3 authored Feb 06, 2018 by Neal Wu Committed by GitHub Feb 06, 2018
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

official/mnist/mnist_tpu.py official/mnist/mnist_tpu.py +1 -1

No files found.
--- a/official/mnist/mnist_tpu.py
+++ b/official/mnist/mnist_tpu.py
@@ -140,7 +140,7 @@ def main(argv):
    tpu_grpc_url = FLAGS.master
  else:
    tpu_cluster_resolver = (
-        tf.contrib.cluster_resolver.python.training.TPUClusterResolver(
+        tf.contrib.cluster_resolver.TPUClusterResolver(
            tpu_names=[FLAGS.tpu_name],
            zone=FLAGS.tpu_zone,
            project=FLAGS.gcp_project))