Fix benchmark naming in reporting

PiperOrigin-RevId: 311003642 Change-Id: I8424106584190b48e5dd7cd6b4031962a1ceeb27
2020-05-11 15:13:30 -07:00 · 2020-05-11 15:13:30 -07:00 · cd6eebfeca
parent ce43a59c72
commit cd6eebfeca
1 changed files with 3 additions and 1 deletions
--- a/tensorflow/python/eager/benchmarks_test_base.py
+++ b/tensorflow/python/eager/benchmarks_test_base.py
@ -32,4 +32,6 @@ class MicroBenchmarksBase(test.Benchmark):
        "examples_per_sec": float("{0:.3f}".format(num_iters / total_time)),
        "us_per_example": float("{0:.3f}".format(total_time * 1e6 / num_iters))
    }
-    self.report_benchmark(iters=num_iters, wall_time=mean_us, extras=extras)
+    benchmark_name = self._get_benchmark_name()
+    self.report_benchmark(
+        iters=num_iters, wall_time=mean_us, extras=extras, name=benchmark_name)