Searched refs:GPUMachineManager (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/ |
| D | devices.cc | 44 se::Platform* gpu_manager = GPUMachineManager(); in GetNumAvailableGPUs()
90 se::Platform* gpu_platform = GPUMachineManager(); in AvailableGPUMemory()
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_process_state.cc | 74 DEVICE_GPU, GPUMachineManager(), tf_gpu_id) in BusIdForGPU()
87 auto executor = DeviceIdUtil::ExecutorForPlatformDeviceId(GPUMachineManager(), in CreateSubAllocator()
144 DeviceIdUtil::CheckValidTfDeviceId(DEVICE_GPU, GPUMachineManager(), in GetGPUAllocator()
225 DeviceIdUtil::CheckValidTfDeviceId(DEVICE_GPU, GPUMachineManager(), in GPUAllocatorCounter()
283 se = DeviceIdUtil::ExecutorForTfDeviceId(DEVICE_GPU, GPUMachineManager(), in GetGpuHostAllocator()
|
| D | gpu_init.h | 36 stream_executor::Platform* GPUMachineManager();
|
| D | gpu_init.cc | 35 se::Platform* GPUMachineManager() { in GPUMachineManager() function
|
| D | gpu_virtual_mem_allocator_test.cc | 40 DeviceIdUtil::ExecutorForPlatformDeviceId(GPUMachineManager(), gpu_id) in CreateAllocator()
53 DeviceIdUtil::ExecutorForPlatformDeviceId(GPUMachineManager(), gpu_id) in TEST()
|
| D | gpu_debug_allocator.cc | 81 stream_exec_ = DeviceIdUtil::ExecutorForPlatformDeviceId(GPUMachineManager(), in GPUDebugAllocator()
160 stream_exec_ = DeviceIdUtil::ExecutorForPlatformDeviceId(GPUMachineManager(), in GPUNanResetAllocator()
|
| D | gpu_cudamalloc_allocator.cc | 32 stream_exec_ = DeviceIdUtil::ExecutorForPlatformDeviceId(GPUMachineManager(), in GPUcudaMallocAllocator()
|
| D | gpu_device.cc | 426 DEVICE_GPU, GPUMachineManager(), tf_gpu_id_); in Init()
783 se::Platform* gpu_manager = GPUMachineManager(); in ParseVisibleDeviceList()
961 GPUMachineManager(), platform_gpu_id) in SingleVirtualDeviceMemoryLimit()
1086 se::Platform* gpu_manager = GPUMachineManager(); in CacheDeviceIds()
1123 se::Platform* gpu_manager = GPUMachineManager(); in GetDeviceDetails()
1147 se::Platform* gpu_manager = GPUMachineManager(); in CreateDevices()
1408 DeviceIdUtil::CheckValidTfDeviceId(DEVICE_GPU, GPUMachineManager(), in CreateGPUDevice()
1415 se::Platform* gpu_manager = GPUMachineManager(); in CreateGPUDevice()
1524 se::Platform* gpu_manager = GPUMachineManager(); in GetDeviceLocalities()
1708 se::Platform* gpu_manager = GPUMachineManager(); in EnablePeerAccess()
[all …]
|
| D | gpu_device_test.cc | 35 DeviceIdUtil::ExecutorForPlatformDeviceId(GPUMachineManager(), gpu_id) in GetTotalGPUMemory()
46 DeviceIdUtil::ExecutorForPlatformDeviceId(GPUMachineManager(), gpu_id) in GetComputeCapability()
182 if (GPUMachineManager()->VisibleDeviceCount() < 2) return; in TEST_F()
|
| D | gpu_debug_allocator_test.cc | 42 return DeviceIdUtil::ExecutorForPlatformDeviceId(GPUMachineManager(), in ExecutorForPlatformGpuId()
|
| D | gpu_bfc_allocator_test.cc | 67 DeviceIdUtil::ExecutorForPlatformDeviceId(GPUMachineManager(), gpu_id) in CreateVirtualMemorySubAllocator()
81 DeviceIdUtil::ExecutorForPlatformDeviceId(GPUMachineManager(), gpu_id) in CreateGPUMemAllocator()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/ |
| D | tensorrt_test.cc | 135 if (CHECK_NOTNULL(GPUMachineManager())->VisibleDeviceCount() <= 0) { in TEST()
|
| /external/tensorflow/tensorflow/core/common_runtime/device/ |
| D | device_event_mgr_test.cc | 112 auto stream_exec = GPUMachineManager()->ExecutorForDevice(0).ValueOrDie(); in TEST()
121 auto stream_exec = GPUMachineManager()->ExecutorForDevice(0).ValueOrDie(); in TEST()
443 auto stream_exec = GPUMachineManager()->ExecutorForDevice(0).ValueOrDie(); in BM_no_ops()
|