Searched refs:gpu_thread_count (Results 1 – 1 of 1) sorted by relevance
418 int64 gpu_thread_count = -1; in Init() local422 ReadInt64FromEnvVar("TF_GPU_THREAD_COUNT", 2, &gpu_thread_count)); in Init()430 static_cast<int32>(gpu_thread_count))); in Init()434 options.env, "gpu_shared", static_cast<int32>(gpu_thread_count)); in Init()