From cd6eebfecab9b93863e7e20acec1ba0481f6b95f Mon Sep 17 00:00:00 2001 From: Yanhua Sun Date: Mon, 11 May 2020 15:13:30 -0700 Subject: [PATCH] Fix benchmark naming in reporting PiperOrigin-RevId: 311003642 Change-Id: I8424106584190b48e5dd7cd6b4031962a1ceeb27 --- tensorflow/python/eager/benchmarks_test_base.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tensorflow/python/eager/benchmarks_test_base.py b/tensorflow/python/eager/benchmarks_test_base.py index 552d844c32d..3d81d08ccbf 100644 --- 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)