tfp.debugging.benchmarking.default_benchmark_config
View source on GitHub
tfp.debugging.benchmarking.default_benchmark_config()