Class WorkerThread

tensorflow/compiler/xla/python/worker_thread.h:31–50

A worker thread that runs a sequence of closures. Equivalent to a thread pool of size 1.

// A worker thread that runs a sequence of closures. Equivalent to a thread
// pool of size 1.
class WorkerThread {
 public:
  // 'name' is a name for the thread for debugging purposes.
  WorkerThread(tensorflow::Env* env, const std::string& name);

  // Blocks until all enqueued closures have completed.
  ~WorkerThread();

  // Adds 'fn' to the queue of closures to be executed by the worker thread.
  void Schedule(std::function<void()> fn);

 private:
  bool WorkAvailable() TF_EXCLUSIVE_LOCKS_REQUIRED(mu_);
  void WorkLoop();

  absl::Mutex mu_;
  std::queue<std::function<void()>> work_queue_ TF_GUARDED_BY(mu_);

  std::unique_ptr<tensorflow::Thread> thread_;
};

}  // namespace xla