tfp.debugging.benchmarking.default_benchmark_config

View source on GitHub