Searched refs:GPUMachineManager (Results 1 – 8 of 8) sorted by relevance
100 auto stream_exec = GPUMachineManager()->ExecutorForDevice(0).ValueOrDie(); in TEST()117 auto stream_exec = GPUMachineManager()->ExecutorForDevice(0).ValueOrDie(); in TEST()150 auto stream_exec = GPUMachineManager()->ExecutorForDevice(0).ValueOrDie(); in TEST()167 auto stream_exec = GPUMachineManager()->ExecutorForDevice(0).ValueOrDie(); in TEST()186 auto stream_exec = GPUMachineManager()->ExecutorForDevice(0).ValueOrDie(); in TEST()208 auto stream_exec = GPUMachineManager()->ExecutorForDevice(0).ValueOrDie(); in TEST()230 auto stream_exec = GPUMachineManager()->ExecutorForDevice(0).ValueOrDie(); in TEST()249 auto stream_exec = GPUMachineManager()->ExecutorForDevice(0).ValueOrDie(); in TEST()
39 return ExecutorForPlatformGpuId(GPUMachineManager(), platform_gpu_id); in ExecutorForPlatformGpuId()53 const int visible_device_count = GPUMachineManager()->VisibleDeviceCount(); in CheckValidTfGpuId()
36 stream_executor::Platform* GPUMachineManager();
35 se::Platform* GPUMachineManager() { in GPUMachineManager() function
36 GpuIdUtil::ExecutorForPlatformGpuId(GPUMachineManager(), gpu_id) in GetTotalGPUMemory()47 GpuIdUtil::ExecutorForPlatformGpuId(GPUMachineManager(), gpu_id) in GetComputeCapability()155 if (GPUMachineManager()->VisibleDeviceCount() < 2) return; in TEST_F()
771 se::Platform* gpu_manager = GPUMachineManager(); in ParseVisibleDeviceList()982 se::Platform* gpu_manager = GPUMachineManager(); in CreateDevices()1515 se::Platform* gpu_manager = GPUMachineManager(); in GetValidDeviceIds()
34 se::Platform* gpu_manager = GPUMachineManager(); in GetNumAvailableGPUs()63 se::Platform* gpu_platform = GPUMachineManager(); in AvailableGPUMemory()
136 if (CHECK_NOTNULL(GPUMachineManager())->VisibleDeviceCount() <= 0) { in TEST()