Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_device.cc1081 if (!cached_device_ids_.empty()) { in CacheDeviceIds()
1098 TF_RETURN_IF_ERROR(GetValidDeviceIds(visible_gpu_order, &cached_device_ids_)); in CacheDeviceIds()
1104 for (PlatformGpuId platform_gpu_id : cached_device_ids_) { in ListPhysicalDevices()
1117 if (device_index < 0 || device_index > cached_device_ids_.size()) { in GetDeviceDetails()
1120 PlatformGpuId platform_gpu_id = cached_device_ids_[device_index]; in GetDeviceDetails()
Dgpu_device.h389 std::vector<PlatformGpuId> cached_device_ids_; variable