Method Run

tensorflow/core/kernels/task_runner.h:34–48

      : func_(f), thread_pool_(tp), num_tasks_(n) {}

  void Run() {
    if (num_tasks_ <= 0) return;
    BlockingCounter bc(num_tasks_ - 1);

    // Sending (num_tasks - 1) tasks to threadpool for scheduling
    for (int32 i = 0; i < num_tasks_ - 1; ++i) {
      thread_pool_->Schedule([this, &bc, i]() {
        func_(i, num_tasks_);
        bc.DecrementCount();
      });
    }
    // Run the last task in current thread.
    func_(num_tasks_ - 1, num_tasks_);
    bc.Wait();
  }

 private:
  std::function<void(int32 task_id, int32 num_tasks)> func_;