diff options
Diffstat (limited to 'tensorflow/tools/docs/parser.py')
-rw-r--r-- | tensorflow/tools/docs/parser.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/tools/docs/parser.py b/tensorflow/tools/docs/parser.py index 51f6adac1e..901b3db22b 100644 --- a/tensorflow/tools/docs/parser.py +++ b/tensorflow/tools/docs/parser.py @@ -586,7 +586,7 @@ def _generate_signature(func, reverse_index): '_ops.GraphKeys': 'tf.GraphKeys', 'init_ops.zeros_initializer': 'tf.zeros_initializer', 'init_ops.ones_initializer': 'tf.ones_initializer', - 'saver_pb2.SaverDef': 'tf.SaverDef', + 'saver_pb2.SaverDef': 'tf.train.SaverDef', } full_name_re = '^%s(.%s)+' % (IDENTIFIER_RE, IDENTIFIER_RE) match = re.match(full_name_re, default_text) |