diff options
| author |  Jingyue Wu <jingyue@google.com> | 2017-03-01 22:39:55 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2017-03-01 22:48:32 -0800 |
| commit | a7c58c8c92a5724cdb530ca1c540cafe549e3f8c (patch) | |
| tree | fcbc2263da4541bfbf3464a45e2e276adda96a7a | |
| parent | 75e45129fc9add331878f88500a45045ca87d439 (diff) | |
Make tf.contrib.memory_stats.MaxBytesInUse accessible.
Change: 148970303
| -rw-r--r-- | tensorflow/contrib/memory_stats/__init__.py | 5 | ||||
| -rw-r--r-- | tensorflow/contrib/memory_stats/python/ops/memory_stats_ops.py | 1 |
2 files changed, 5 insertions, 1 deletions
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() |