Home
last modified time | relevance | path

Searched refs:visible_gpu_id (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_device.cc1299 int visible_gpu_id = visible_gpu_order[i].value(); in CreateDevices() local
1302 if (visible_gpu_initialized_[visible_gpu_id]) { in CreateDevices()
1306 visible_gpu_initialized_[visible_gpu_id] = true; in CreateDevices()
1760 int visible_gpu_id = visible_gpu_order[i].value(); in GetDefaultMinGPUMultiprocessorCount() local
1761 auto description_status = gpu_manager->DescriptionForDevice(visible_gpu_id); in GetDefaultMinGPUMultiprocessorCount()
1894 int visible_gpu_id = visible_gpu_order[i].value(); in GetValidDeviceIds() local
1895 auto description_status = gpu_manager->DescriptionForDevice(visible_gpu_id); in GetValidDeviceIds()
1960 const PlatformDeviceId visible_gpu_id = visible_gpu_order[i]; in GetValidDeviceIds() local
1962 gpu_manager->DescriptionForDevice(visible_gpu_id.value()); in GetValidDeviceIds()
1964 LOG(INFO) << "Ignoring visible gpu device " << visible_gpu_id in GetValidDeviceIds()
[all …]