Searched refs:min_gpu_core_count (Results 1 – 2 of 2) sorted by relevance
42 int min_gpu_core_count = 8; in GetNumAvailableGPUs() local43 if (desc.core_count() >= min_gpu_core_count) { in GetNumAvailableGPUs()
1388 int min_gpu_core_count = -1; in GetMinGPUMultiprocessorCount() local1389 if (strings::safe_strto32(tf_min_gpu_core_count, &min_gpu_core_count)) { in GetMinGPUMultiprocessorCount()1390 if (min_gpu_core_count >= 0) { in GetMinGPUMultiprocessorCount()1391 return min_gpu_core_count; in GetMinGPUMultiprocessorCount()1598 int min_gpu_core_count = in GetValidDeviceIds() local1658 if (desc.core_count() < min_gpu_core_count) { in GetValidDeviceIds()1663 << ". The minimum required count is " << min_gpu_core_count in GetValidDeviceIds()