Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_debug_allocator_test.cc39 GpuIdUtil::ExecutorForPlatformGpuId(platform_gpu_id).ValueOrDie(), in TEST()
44 GpuIdUtil::ExecutorForPlatformGpuId(platform_gpu_id).ValueOrDie(); in TEST()
67 GpuIdUtil::ExecutorForPlatformGpuId(platform_gpu_id).ValueOrDie(), in TEST()
72 GpuIdUtil::ExecutorForPlatformGpuId(platform_gpu_id).ValueOrDie(); in TEST()
103 GpuIdUtil::ExecutorForPlatformGpuId(platform_gpu_id).ValueOrDie(), in TEST()
108 GpuIdUtil::ExecutorForPlatformGpuId(platform_gpu_id).ValueOrDie(); in TEST()
136 GpuIdUtil::ExecutorForPlatformGpuId(platform_gpu_id).ValueOrDie(), in TEST()
141 GpuIdUtil::ExecutorForPlatformGpuId(platform_gpu_id).ValueOrDie(); in TEST()
181 GpuIdUtil::ExecutorForPlatformGpuId(platform_gpu_id).ValueOrDie(), in TEST()
188 GpuIdUtil::ExecutorForPlatformGpuId(platform_gpu_id).ValueOrDie(); in TEST()
[all …]
Dgpu_bfc_allocator_test.cc55 GpuIdUtil::ExecutorForPlatformGpuId(platform_gpu_id).ValueOrDie(), in TEST()
88 GpuIdUtil::ExecutorForPlatformGpuId(platform_gpu_id).ValueOrDie(), in TEST()
150 GpuIdUtil::ExecutorForPlatformGpuId(platform_gpu_id).ValueOrDie(), in TEST()
189 GpuIdUtil::ExecutorForPlatformGpuId(platform_gpu_id).ValueOrDie(), in TEST()
199 GpuIdUtil::ExecutorForPlatformGpuId(platform_gpu_id).ValueOrDie(), in TEST()
208 GpuIdUtil::ExecutorForPlatformGpuId(platform_gpu_id).ValueOrDie(), in TEST()
220 GpuIdUtil::ExecutorForPlatformGpuId(platform_gpu_id).ValueOrDie(), in TEST()
240 GpuIdUtil::ExecutorForPlatformGpuId(platform_gpu_id).ValueOrDie(), in TEST()
306 GpuIdUtil::ExecutorForPlatformGpuId(platform_gpu_id).ValueOrDie(), in TEST()
310 GpuIdUtil::ExecutorForPlatformGpuId(platform_gpu_id).ValueOrDie(), in TEST()
[all …]
Dgpu_id_utils.h33 static se::port::StatusOr<se::StreamExecutor*> ExecutorForPlatformGpuId( in ExecutorForPlatformGpuId() function
37 static se::port::StatusOr<se::StreamExecutor*> ExecutorForPlatformGpuId( in ExecutorForPlatformGpuId() function
39 return ExecutorForPlatformGpuId(GPUMachineManager(), platform_gpu_id); in ExecutorForPlatformGpuId()
46 return ExecutorForPlatformGpuId(platform_gpu_id); in ExecutorForTfGpuId()
Dgpu_device.cc893 GpuIdUtil::ExecutorForPlatformGpuId(platform_gpu_id).ValueOrDie(); in SingleVirtualDeviceMemoryLimit()
1196 GpuIdUtil::ExecutorForPlatformGpuId(platform_gpu_id).ValueOrDie(); in CreateGPUDevice()
1240 GpuIdUtil::ExecutorForPlatformGpuId(platform, platform_gpu_i) in GetPeerAccessMap()
1243 GpuIdUtil::ExecutorForPlatformGpuId(platform, platform_gpu_j) in GetPeerAccessMap()
1291 GpuIdUtil::ExecutorForPlatformGpuId(platform_gpu_id).ValueOrDie(); in GetDeviceLocalities()
1361 GpuIdUtil::ExecutorForPlatformGpuId(gpu_manager, visible_gpu_order[i]); in GetDefaultMinGPUMultiprocessorCount()
1477 GpuIdUtil::ExecutorForPlatformGpuId(platform, platform_gpu_i) in EnablePeerAccess()
1480 GpuIdUtil::ExecutorForPlatformGpuId(platform, platform_gpu_j) in EnablePeerAccess()
1529 GpuIdUtil::ExecutorForPlatformGpuId(gpu_manager, visible_gpu_id); in GetValidDeviceIds()
1605 GpuIdUtil::ExecutorForPlatformGpuId(gpu_manager, visible_gpu_id); in GetValidDeviceIds()
Dgpu_debug_allocator.cc82 GpuIdUtil::ExecutorForPlatformGpuId(platform_gpu_id).ValueOrDie(); in GPUDebugAllocator()
160 GpuIdUtil::ExecutorForPlatformGpuId(platform_gpu_id).ValueOrDie(); in GPUNanResetAllocator()
Dgpu_cudamalloc_allocator.cc34 GpuIdUtil::ExecutorForPlatformGpuId(platform_gpu_id).ValueOrDie(); in GPUcudaMallocAllocator()
Dgpu_device_test.cc36 GpuIdUtil::ExecutorForPlatformGpuId(GPUMachineManager(), gpu_id) in GetTotalGPUMemory()
47 GpuIdUtil::ExecutorForPlatformGpuId(GPUMachineManager(), gpu_id) in GetComputeCapability()
Dgpu_process_state.cc109 GpuIdUtil::ExecutorForPlatformGpuId(platform_gpu_id).ValueOrDie(), in GetGPUAllocator()