Function LookupDevice

tensorflow/compiler/xla/python/local_client.cc:544–550

}

static std::shared_ptr<Device> LookupDevice(const PyLocalClient& client,
                                            int device_id) {
  auto it = client.id_to_device().find(device_id);
  CHECK(it != client.id_to_device().end())
      << "Unknown device id: " << device_id;
  return it->second;
}

PyLocalExecutable::PyLocalExecutable(
    std::vector<std::unique_ptr<LocalExecutable>> executables,