Searched refs:gpu_thread_mode (Results 1 – 1 of 1) sorted by relevance
413 string gpu_thread_mode; in Init() local415 ReadStringFromEnvVar("TF_GPU_THREAD_MODE", "global", &gpu_thread_mode)); in Init()416 gpu_thread_mode = str_util::Lowercase(gpu_thread_mode); in Init()417 if (gpu_thread_mode != "global") { in Init()423 if (gpu_thread_mode == "gpu_private") { in Init()432 } else if (gpu_thread_mode == "gpu_shared") { in Init()438 strings::StrCat("Invalid gpu_thread_mode: ", gpu_thread_mode); in Init()