Searched defs:platform_gpu_id (Results 1 – 13 of 13) sorted by relevance
48 const PlatformGpuId platform_gpu_id(0); in TEST() local144 const PlatformGpuId platform_gpu_id(0); in TEST() local186 const PlatformGpuId platform_gpu_id(0); in TEST() local231 const PlatformGpuId platform_gpu_id(0); in TEST() local241 const PlatformGpuId platform_gpu_id(0); in TEST() local
24 TfDeviceId tf_gpu_id, PlatformDeviceId platform_gpu_id) { in InsertTfPlatformGpuIdPair()30 PlatformDeviceId* platform_gpu_id) { in TfToPlatformGpuId()
135 PlatformGpuId platform_gpu_id; in Reinitialize() local476 PlatformGpuId platform_gpu_id; in Init() local797 int32 platform_gpu_id; in ParseVisibleDeviceList() local956 PlatformGpuId platform_gpu_id, in SingleVirtualDeviceMemoryLimit()1104 for (PlatformGpuId platform_gpu_id : cached_device_ids_) { in ListPhysicalDevices() local1120 PlatformGpuId platform_gpu_id = cached_device_ids_[device_index]; in GetDeviceDetails() local1216 for (PlatformGpuId platform_gpu_id : valid_platform_gpu_ids) { in CreateDevices() local1336 const PlatformGpuId platform_gpu_id = valid_platform_gpu_ids[i]; in CreateDevices() local1379 static string GetShortDeviceDescription(PlatformGpuId platform_gpu_id, in GetShortDeviceDescription()1410 PlatformGpuId platform_gpu_id; in CreateGPUDevice() local[all …]
30 PlatformGpuId platform_gpu_id) in GPUcudaMallocAllocator()
84 const GPUOptions& options, PlatformGpuId platform_gpu_id, in CreateSubAllocator()159 PlatformGpuId platform_gpu_id; in GetGPUAllocator() local
79 PlatformGpuId platform_gpu_id) in GPUDebugAllocator()158 PlatformGpuId platform_gpu_id) in GPUNanResetAllocator()
90 PlatformGpuId platform_gpu_id; in gpu_id() local
77 DeviceProperties GetLocalGPUInfo(PlatformGpuId platform_gpu_id) { in GetLocalGPUInfo()160 PlatformGpuId platform_gpu_id; in GetDeviceInfo() local
96 PlatformGpuId platform_gpu_id; in Provision() local
47 int32 platform_gpu_id; in ParseVisibleDeviceList() local
108 PlatformGpuId platform_gpu_id; in GetFirstValidDeviceId() local270 PlatformGpuId platform_gpu_id; in GetEngineInfo() local639 PlatformGpuId platform_gpu_id; in GetDeviceAndAllocator() local
245 PlatformGpuId platform_gpu_id; in GetDeviceInfo() local
1047 const int platform_gpu_id = in AllocateCalibrationResources() local