Searched refs:_is_gpu_device (Results 1 – 1 of 1) sorted by relevance
131 def _is_gpu_device(device): function153 return sum(1 for d in devices if _is_gpu_device(d))160 num_gpus = sum(1 for d in device_in_task if _is_gpu_device(d))163 elif num_gpus != sum(1 for d in device_in_task if _is_gpu_device(d)):