Skip to content

GitLab

Commit a7bf2b41 authored by Denali Molitor's avatar Denali Molitor Committed by A. Unique TensorFlower
Browse files

Add transformer and BERT benchmarks for Tensor Tracer. 

PiperOrigin-RevId: 443751793
parent 7dbedf73
Hide whitespace changes
Inline Side-by-side
Showing with 6 additions and 1 deletion
official/benchmark/base_benchmark.py
View file @ a7bf2b41
...@@ -73,7 +73,9 @@ class BaseBenchmark( # pylint: disable=undefined-variable ...@@ -73,7 +73,9 @@ class BaseBenchmark( # pylint: disable=undefined-variable
_benchmark_parameters = _get_benchmark_params( _benchmark_parameters = _get_benchmark_params(
benchmark_definitions.VISION_BENCHMARKS) + _get_benchmark_params( benchmark_definitions.VISION_BENCHMARKS) + _get_benchmark_params(
benchmark_definitions.NLP_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, def __init__(self,
output_dir=None, output_dir=None,
......
official/benchmark/benchmark_definitions.py
View file @ a7bf2b41
...@@ -53,3 +53,6 @@ NLP_BENCHMARKS = { ...@@ -53,3 +53,6 @@ NLP_BENCHMARKS = {
QAT_BENCHMARKS = { QAT_BENCHMARKS = {
} }
TENSOR_TRACER_BENCHMARKS = {
}
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment