Internal change
PiperOrigin-RevId: 406400290
Change-Id: I109c3747514a3607372fb22cb5a5226b53dca31a
diff --git a/tensorflow/python/eager/run_eager_op_as_function_test.py b/tensorflow/python/eager/run_eager_op_as_function_test.py
index e5d7d77..ccc96cc 100644
--- a/tensorflow/python/eager/run_eager_op_as_function_test.py
+++ b/tensorflow/python/eager/run_eager_op_as_function_test.py
@@ -83,7 +83,7 @@
return name
def _run(self, func, num_iters):
- self.run_report(run_benchmark, func, num_iters)
+ self.run_report(run_benchmark, func, num_iters, report_mean_us=True)
def _benchmark_matmul(self, mat, device):
if device == GPU and not context.num_gpus():