tfp.debugging.benchmarking.default_benchmark_config

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