Add transformer and BERT benchmarks for Tensor Tracer.

PiperOrigin-RevId: 443751793

Add transformer and BERT benchmarks for Tensor Tracer.
PiperOrigin-RevId: 443751793
a7bf2b41 · Denali Molitor · A. Unique TensorFlower · 7dbedf73 · a7bf2b41 · a7bf2b41
Commit a7bf2b41 authored Apr 22, 2022 by Denali Molitor Committed by A. Unique TensorFlower Apr 22, 2022
Show whitespace changes
Inline Side-by-side

Showing with 6 additions and 1 deletion

official/benchmark/base_benchmark.py official/benchmark/base_benchmark.py +3 -1

official/benchmark/benchmark_definitions.py official/benchmark/benchmark_definitions.py +3 -0

No files found.
--- a/official/benchmark/base_benchmark.py
+++ b/official/benchmark/base_benchmark.py
@@ -73,7 +73,9 @@ class BaseBenchmark(  # pylint: disable=undefined-variable
  _benchmark_parameters = _get_benchmark_params(
      benchmark_definitions.VISION_BENCHMARKS) + _get_benchmark_params(
          benchmark_definitions.NLP_BENCHMARKS) + _get_benchmark_params(
-              benchmark_definitions.QAT_BENCHMARKS, True)
+              benchmark_definitions.QAT_BENCHMARKS,
+              True) + _get_benchmark_params(
+                  benchmark_definitions.TENSOR_TRACER_BENCHMARKS)

  def __init__(self,
               output_dir=None,

--- a/official/benchmark/benchmark_definitions.py
+++ b/official/benchmark/benchmark_definitions.py
@@ -53,3 +53,6 @@ NLP_BENCHMARKS = {

 QAT_BENCHMARKS = {
 }
+
+TENSOR_TRACER_BENCHMARKS = {
+}