Searched refs:gpu_thread_count (Results 1 – 1 of 1) sorted by relevance
496 int64 gpu_thread_count = -1; in Init() local500 ReadInt64FromEnvVar("TF_GPU_THREAD_COUNT", 2, &gpu_thread_count)); in Init()509 static_cast<int32>(gpu_thread_count), in Init()516 static_cast<int32>(gpu_thread_count), in Init()