In #2276, auto_parallel(metagraph, model) try to import google3

This will error when num_gpus > 1

In #2276, auto_parallel(metagraph, model) try to import google3
This will error when num_gpus > 1
3d6dc1dd · Yen-chi Chen · 09a32f32 · 3d6dc1dd
Commit 3d6dc1dd authored Sep 01, 2017 by Yen-chi Chen
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

tutorials/rnn/ptb/util.py tutorials/rnn/ptb/util.py +1 -1

No files found.
--- a/tutorials/rnn/ptb/util.py
+++ b/tutorials/rnn/ptb/util.py
@@ -86,7 +86,7 @@ class UpdateCollection(object):


 def auto_parallel(metagraph, model):
-  from google3.third_party.tensorflow.python.grappler import tf_optimizer
+  from tensorflow.python.grappler import tf_optimizer
  rewriter_config = rewriter_config_pb2.RewriterConfig()
  rewriter_config.optimizers.append("autoparallel")
  rewriter_config.auto_parallel.enable = True