From a7c58c8c92a5724cdb530ca1c540cafe549e3f8c Mon Sep 17 00:00:00 2001 From: Jingyue Wu Date: Wed, 1 Mar 2017 22:39:55 -0800 Subject: Make tf.contrib.memory_stats.MaxBytesInUse accessible. Change: 148970303 --- tensorflow/contrib/memory_stats/__init__.py | 5 ++++- tensorflow/contrib/memory_stats/python/ops/memory_stats_ops.py | 1 + 2 files changed, 5 insertions(+), 1 deletion(-) diff --git a/tensorflow/contrib/memory_stats/__init__.py b/tensorflow/contrib/memory_stats/__init__.py index 78146b701e..e310f9fd58 100644 --- a/tensorflow/contrib/memory_stats/__init__.py +++ b/tensorflow/contrib/memory_stats/__init__.py @@ -12,7 +12,10 @@ # See the License for the specific language governing permissions and # limitations under the License. # ============================================================================== -"""Ops for memory statistics.""" +"""Ops for memory statistics. + +@@MaxBytesInUse +""" from tensorflow.contrib.memory_stats.python.ops.memory_stats_ops import MaxBytesInUse diff --git a/tensorflow/contrib/memory_stats/python/ops/memory_stats_ops.py b/tensorflow/contrib/memory_stats/python/ops/memory_stats_ops.py index 0e0772e6be..60472406a3 100644 --- a/tensorflow/contrib/memory_stats/python/ops/memory_stats_ops.py +++ b/tensorflow/contrib/memory_stats/python/ops/memory_stats_ops.py @@ -27,4 +27,5 @@ _memory_stats_ops_so = loader.load_op_library( def MaxBytesInUse(): + """Generates an op that computes the peak memory of a device.""" return gen_memory_stats_ops.max_bytes_in_use() -- cgit v1.2.3