Method RunRestoreOps

serving/processor/serving/model_session.cc:187–230

}

Status ModelSessionMgr::RunRestoreOps(
    const char* ckpt_name, int64 full_ckpt_version,
    const char* savedmodel_dir, Session* session,
    IFeatureStoreMgr* sparse_storage,
    bool is_incr_ckpt, bool update_sparse,
    int64_t latest_version) {
  std::vector<std::pair<std::string, Tensor>> extra_tensors;
  Tensor sparse_storage_tensor(DT_UINT64, TensorShape({}));
  sparse_storage_tensor.scalar<uint64>()() =
      reinterpret_cast<uint64>(sparse_storage);
  auto sparse_storage_tensor_pair = std::make_pair(
      GetStoragePointerNodeName(), sparse_storage_tensor);
  extra_tensors.emplace_back(sparse_storage_tensor_pair);

  Tensor version_tensor(DT_UINT64, TensorShape({}));
  version_tensor.scalar<uint64>()() = full_ckpt_version;
  auto version_tensor_pair = std::make_pair(
      GetModelVersionNodeName(), version_tensor);
  extra_tensors.emplace_back(version_tensor_pair);

  Tensor is_incr_ckpt_tensor(DT_BOOL, TensorShape({}));
  is_incr_ckpt_tensor.scalar<bool>()() = is_incr_ckpt;
  auto is_incr_ckpt_tensor_pair = std::make_pair(
      GetIncrCkptNodeName(), is_incr_ckpt_tensor);
  extra_tensors.emplace_back(is_incr_ckpt_tensor_pair);

  TF_RETURN_IF_ERROR(
      util::RunRestore(*run_options_, ckpt_name, savedmodel_dir,
          meta_graph_def_.saver_def().restore_op_name(),
          meta_graph_def_.saver_def().filename_tensor_name(),
          asset_file_defs_, session, update_sparse,
          latest_version, extra_tensors));

  if (util::HasMainOp(meta_graph_def_)) {
    return util::RunMainOp(*run_options_, savedmodel_dir,
        meta_graph_def_, asset_file_defs_, session,
        kSavedModelMainOpKey, sparse_storage_tensor_pair);
  } else {
    return util::RunMainOp(
        *run_options_, savedmodel_dir, meta_graph_def_,
        asset_file_defs_, session, kSavedModelLegacyInitOpKey,
        sparse_storage_tensor_pair);
  }
}

ModelSession::ModelSession(SessionGroup* s,
    const std::string& select_session_policy,