PY3 migration - Remove xrange in hvp_test.py.

PiperOrigin-RevId: 286311054 Change-Id: I1225b7ed8f6ca83368dac2bb1572b6abc7945fd3
2019-12-18 19:16:07 -08:00 · 2019-12-18 19:16:07 -08:00 · afd8b4dbcf
commit afd8b4dbcf
parent e1d2b42661
2 changed files with 2 additions and 0 deletions
--- a/tensorflow/python/eager/benchmarks/resnet50/BUILD
+++ b/tensorflow/python/eager/benchmarks/resnet50/BUILD
@ -71,6 +71,7 @@ cuda_py_test(
        ":resnet50_test_util",
        "//tensorflow:tensorflow_py_no_contrib",
        "//tensorflow/python/eager:forwardprop",
+        "@six_archive//:six",
    ],
 )

--- a/tensorflow/python/eager/benchmarks/resnet50/hvp_test.py
+++ b/tensorflow/python/eager/benchmarks/resnet50/hvp_test.py
@ -22,6 +22,7 @@ import gc
 import time

 from absl.testing import parameterized
+from six.moves import xrange
 import tensorflow as tf

 from tensorflow.python.eager import forwardprop