线程池ThreadPool

星空探索

于 2017-12-18 09:15:29 发布

阅读量327

点赞数

CC 4.0 BY-SA版权

分类专栏： Tensorflow研究学习

本文链接：https://blog.youkuaiyun.com/sunlei0625/article/details/78829672

Tensorflow研究学习专栏收录该内容

5 篇文章

订阅专栏

本文详细介绍了TensorFlow中线程池类的定义及使用方法。主要涉及线程池构造函数的不同形式及其参数意义，如环境配置、线程选项、线程数量等。此外还阐述了线程池的主要成员函数，包括任务调度、并行循环等功能。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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);
};

}
}