diff options
Diffstat (limited to 'tensorflow/python/eager/memory_test.py')
| -rw-r--r-- | tensorflow/python/eager/memory_test.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/tensorflow/python/eager/memory_test.py b/tensorflow/python/eager/memory_test.py index 74c6cbdd31..a1a59d511f 100644 --- a/tensorflow/python/eager/memory_test.py +++ b/tensorflow/python/eager/memory_test.py @@ -24,6 +24,8 @@ from __future__ import absolute_import from __future__ import division from __future__ import print_function +import six + from tensorflow.python import keras from tensorflow.python.eager import backprop from tensorflow.python.eager import context @@ -63,7 +65,7 @@ class MemoryTest(test.TestCase): initial = memory_profiler.memory_usage(-1)[0] - for _ in xrange(num_iters): + for _ in six.moves.range(num_iters): f() increase = memory_profiler.memory_usage(-1)[0] - initial |