Searched refs:gpu_platform (Results 1 – 1 of 1) sorted by relevance
90 se::Platform* gpu_platform = GPUMachineManager(); in AvailableGPUMemory() local91 CHECK_LT(gpu_id, gpu_platform->VisibleDeviceCount()); in AvailableGPUMemory()92 se::StreamExecutor* se = gpu_platform->ExecutorForDevice(gpu_id).ValueOrDie(); in AvailableGPUMemory()