线程池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);};

}}