Method ThenExecute

tensorflow/core/common_runtime/gpu/gpu_event_mgr.h:72–76

Execute func when all pending stream actions have completed. func must be brief and non-blocking since it executes in the one thread used for all such callbacks and also buffer deletions.

  // func must be brief and non-blocking since it executes in the one
  // thread used for all such callbacks and also buffer deletions.
  inline void ThenExecute(se::Stream* stream, std::function<void()> func) {
    mutex_lock l(mu_);
    QueueFunc(stream, std::move(func));
    PollEvents(stream);
  }

 private:
  friend class TEST_EventMgr;