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