diff options
| author |  Anna R <annarev@google.com> | 2018-02-09 11:16:01 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-02-09 11:21:21 -0800 |
| commit | 6c85a66a16f07bab9b5dc3df33bc6b8111b76615 (patch) | |
| tree | 53b3f02472af628d4b42904962bf99e9c1e2b837 /tensorflow/python/profiler | |
| parent | 249b67204e5c1eb4805b8b021386aa0364045063 (diff) | |
Add export calls for protos.
PiperOrigin-RevId: 185166764
Diffstat (limited to 'tensorflow/python/profiler')
| -rw-r--r-- | tensorflow/python/profiler/profiler.py | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/tensorflow/python/profiler/profiler.py b/tensorflow/python/profiler/profiler.py index 130dcb5134..fa7f30b236 100644 --- a/tensorflow/python/profiler/profiler.py +++ b/tensorflow/python/profiler/profiler.py @@ -31,6 +31,7 @@ from tensorflow.python.profiler.option_builder import ProfileOptionBuilder from tensorflow.python.profiler.tfprof_logger import write_op_log from tensorflow.python.util.all_util import remove_undocumented +from tensorflow.python.util.tf_export import tf_export _allowed_symbols = [ @@ -48,6 +49,12 @@ _allowed_symbols.extend([ 'OpLogProto', ]) +# Export protos +tf_export('profiler.GraphNodeProto')(GraphNodeProto) +tf_export('profiler.MultiGraphNodeProto')(MultiGraphNodeProto) +tf_export('profiler.AdviceProto')(AdviceProto) +tf_export('profiler.OpLogProto')(OpLogProto) + remove_undocumented(__name__, _allowed_symbols, [ Profiler, profile, |