aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/memory_stats/__init__.py
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/contrib/memory_stats/__init__.py')
-rw-r--r--tensorflow/contrib/memory_stats/__init__.py4
1 files changed, 4 insertions, 0 deletions
diff --git a/tensorflow/contrib/memory_stats/__init__.py b/tensorflow/contrib/memory_stats/__init__.py
index a32302c854..2ce849ca66 100644
--- a/tensorflow/contrib/memory_stats/__init__.py
+++ b/tensorflow/contrib/memory_stats/__init__.py
@@ -19,6 +19,10 @@
@@MaxBytesInUse
"""
+from __future__ import absolute_import
+from __future__ import division
+from __future__ import print_function
+
from tensorflow.contrib.memory_stats.python.ops.memory_stats_ops import BytesInUse
from tensorflow.contrib.memory_stats.python.ops.memory_stats_ops import BytesLimit
from tensorflow.contrib.memory_stats.python.ops.memory_stats_ops import MaxBytesInUse