Don't wait for session.close() to finish when calculating benchmark t…

…iming. PiperOrigin-RevId: 206634868
willsim · Jul 30, 2018 · cfd9cc4 · cfd9cc4
1 parent 53fcaa2
commit cfd9cc4
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tensorflow/python/kernel_tests/decode_jpeg_op_test.py b/tensorflow/python/kernel_tests/decode_jpeg_op_test.py
@@ -110,7 +110,8 @@ def _evalDecodeJpeg(self,
       start_time = time.time()
       for _ in xrange(num_iters):
         sess.run(r)
-    return time.time() - start_time
+      end_time = time.time()
+    return end_time - start_time
 
   def benchmarkDecodeJpegSmall(self):
     """Evaluate single DecodeImageOp for small size image."""