Function GPUMachineManager

tensorflow/core/common_runtime/gpu/gpu_init.cc:35–43

}

se::Platform* GPUMachineManager() {
  auto result = se::MultiPlatformManager::PlatformWithName(GpuPlatformName());
  if (!result.ok()) {
    LOG(FATAL) << "Could not find Platform with name " << GpuPlatformName();
    return nullptr;
  }

  return result.ValueOrDie();
}

string GpuPlatformName() {
#if TENSORFLOW_USE_ROCM