experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Log value of TF_AUTOTUNE_THRESHOLD

Browse Source 
PiperOrigin-RevId: 341097977
Change-Id: Ida4067d3076c76ff86ca40f108ef3f408b91c1d3
This commit is contained in:
Sanjoy Das 2020-11-06 12:28:20 -08:00 committed by TensorFlower Gardener
parent 3d28521352
commit dce6d63144
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tensorflow/core/kernels/gpu_utils.h
View File

@ -151,6 +151,7 @@ class AutoTuneMap {
int min_warmup_iterations = 10;
const char* threshold_str = getenv("TF_AUTOTUNE_THRESHOLD");
if (threshold_str != nullptr) {
VLOG(1) << "TF_AUTOTUNE_THRESHOLD = " << threshold_str;
strings::safe_strto32(threshold_str, &min_score_threshold_);
}
const char* min_warmup_iteration_str =