Fix benchmark naming in reporting

PiperOrigin-RevId: 311003642
Change-Id: I8424106584190b48e5dd7cd6b4031962a1ceeb27
This commit is contained in:
Yanhua Sun 2020-05-11 15:13:30 -07:00 committed by TensorFlower Gardener
parent ce43a59c72
commit cd6eebfeca
1 changed files with 3 additions and 1 deletions

View File

@ -32,4 +32,6 @@ class MicroBenchmarksBase(test.Benchmark):
"examples_per_sec": float("{0:.3f}".format(num_iters / total_time)), "examples_per_sec": float("{0:.3f}".format(num_iters / total_time)),
"us_per_example": float("{0:.3f}".format(total_time * 1e6 / num_iters)) "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)