Function RestoreHashTable

tensorflow/core/kernels/save_restore_tensor.cc:569–634

}

void RestoreHashTable(
    std::function<void(std::function<void()>)> runner,
    BundleReader* reader, HashTable* table,
    const std::vector<TensibleVariable*>& tensibles,
    const string& table_name, const std::vector<string>& tensibles_name,
    int64 slice_beg, int64 slice_length, int64 slice_size,
    std::function<void(Status)> done) {
  LOG(INFO) << "Restore";
  table->Clear([=](Status st) {
    if (!st.ok()) {
      done(st);
      return;
    }
    std::vector<TensorSliceProto> table_slices;
    std::vector<std::vector<TensorSliceProto>> tensible_slices(tensibles.size());
    if (tensibles.size() != tensibles_name.size()) {
      done(errors::FailedPrecondition("Tensible Size is not equal to names"));
      return;
    }

    Status res = LookupSliceInfo(
        reader, table_name, tensibles_name, &table_slices, &tensible_slices);
    if (!res.ok()) {
      done(res);
      return;
    }

    std::vector<RestoreHashTableSlice> slices;
    BuildRestoreSlice(table_slices, slice_beg, slice_length, &slices);

    mutex* mu = new mutex;
    auto insert_table = std::bind(
        InsertTable, reader, table, tensibles, table_name,
        tensibles_name, table_slices, tensible_slices, std::placeholders::_1,
        mu, slice_beg, slice_length, slice_size, nullptr);

    auto after_add_table = [=](Status st) {
      delete mu;
      if (!st.ok()) {
        done(st);
        return;
      };
      std::set<TensibleVariable*> restored_variable(
          tensibles.begin(), tensibles.end());
      std::vector<TensibleVariable*> vars = table->Tensibles();
      StatusCollector* stc = new StatusCollector(vars.size(), done);
      for (auto&& tensible : vars) {
        if (restored_variable.find(tensible) != restored_variable.end()) {
          tensible->Pad(table->Size(), stc->AddStatusFunc());
        } else {
          tensible->Resize(table->Size(), stc->AddStatusFunc());
        }
      }
      LOG(INFO) << "Restore Done";
      stc->Start();
    };
    StatusCollector* stc = new StatusCollector(slices.size(), after_add_table);
    for (auto s : slices) {