Method Schedule

tensorflow/core/kernels/work_queue_ops.cc:169–184

  }

  void Schedule(int64 num_threads, std::function<void()> fn) {
    std::unique_lock<std::mutex> lock(mu_);
    if (threads_) {
      lock.unlock();
      threads_->Schedule(fn);
      return;
    }

    threads_.reset(
        new thread::ThreadPool(Env::Default(), ThreadOptions(),
                               strings::StrCat("work_queue_threads_", name_),
                               num_threads, false /* low_latency_hint */));

    lock.unlock();
    threads_->Schedule(fn);
  }

 private:
  // TODO(yuanman.ym): Use memory efficient data structure, e.g. HAT-trie,