线程池ThreadPool

tensorflow::tensorflow/core/lib/core/threadpool.h
 
 
namespace tensorflow {
namespace thread {

 
class ThreadPool {
 public:
  ThreadPool(Env* env, const ThreadOptions& thread_options, const string& name,
             int num_threads, bool low_latency_hint);
  ThreadPool(Env* env, const string& name, int num_threads);
 
  ThreadPool(Env* env, const ThreadOptions& thread_options, const string& name,
             int num_threads);
 
  ~ThreadPool();
 
  void Schedule(std::function<void()> fn);
 
  void ParallelFor(int64 total, int64 cost_per_unit,
                   std::function<void(int64, int64)> fn);
 
  void ParallelForWithWorkerId(
      int64 total, int64 cost_per_unit,
      const std::function<void(int64, int64, int)>& fn);
 
  int NumThreads() const;
 
  int CurrentThreadId() const;
 
  struct Impl;
 
 private:
  std::unique_ptr<Impl> impl_;
  TF_DISALLOW_COPY_AND_ASSIGN(ThreadPool);
};
 
}
}