Skip to content

GitLab

Commit 55e6e387 authored by Toby Boyd's avatar Toby Boyd Committed by GitHub
Browse files

Merge pull request #2321 from mark86092/ptb-fix 

Fix import error in RNN ptb tutorial
parents 64f0ead2 3d6dc1dd
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tutorials/rnn/ptb/util.py
View file @ 55e6e387
...@@ -86,7 +86,7 @@ class UpdateCollection(object): ...@@ -86,7 +86,7 @@ class UpdateCollection(object):
def auto_parallel(metagraph, model): 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 = rewriter_config_pb2.RewriterConfig()
rewriter_config.optimizers.append("autoparallel") rewriter_config.optimizers.append("autoparallel")
rewriter_config.auto_parallel.enable = True rewriter_config.auto_parallel.enable = True
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment