Method Update

tensorflow/core/kernels/incr_save_restore_ops.h:94–113

  }

  void Update(const Tensor& indices, OpKernelContext *ctx) {
    const int64 N = indices.NumElements();
    auto thread_pool = *(ctx->device()->tensorflow_cpu_worker_threads());

    std::vector<std::pair<int64, int64>> parts;
    SplitParallelParts(N,
        std::min(part_count_, thread_pool.workers->NumThreads()), parts);

    int part_count = parts.size();
    BlockingCounter counter(part_count);
    for (int i = 0; i < part_count; i++) {
      int64 start = parts[i].first;
      int64 end = parts[i].second;
      thread_pool.workers->Schedule([this, indices, i, start, end, &counter]() {
          hash_maps_[i].Update(indices, start, end);
          counter.DecrementCount();
        });
    }
    counter.Wait();
  }

  void Swap(std::unordered_map<T, uint64> &indices) {
    std::vector<std::unordered_map<T, uint64>> tmp_maps;