diff options
| -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() |