Add non-functional stubs for a closure executor to tensorflow::Env.

Change: 110797223

1 files changed, 17 insertions, 0 deletions

diff --git a/tensorflow/core/public/env.h b/tensorflow/core/public/env.h
index fe68167c04..ac34a02c89 100644
--- a/tensorflow/core/public/env.h
+++ b/tensorflow/core/public/env.h
@@ -134,6 +134,17 @@ class Env {
                               const string& name,
                               std::function<void()> fn) TF_MUST_USE_RESULT = 0;
 
+  // \brief Schedules the given closure on a thread-pool.
+  //
+  // NOTE(mrry): This closure must not block.
+  virtual void SchedClosure(std::function<void()> closure) = 0;
+
+  // \brief Schedules the given closure on a thread-pool after the given number
+  // of microseconds.
+  //
+  // NOTE(mrry): This closure must not block.
+  virtual void SchedClosureAfter(int micros, std::function<void()> closure) = 0;
+
  private:
   /// No copying allowed
   Env(const Env&);
@@ -234,6 +245,12 @@ class EnvWrapper : public Env {
                       std::function<void()> fn) override {
     return target_->StartThread(thread_options, name, fn);
   }
+  void SchedClosure(std::function<void()> closure) override {
+    target_->SchedClosure(closure);
+  }
+  void SchedClosureAfter(int micros, std::function<void()> closure) override {
+    target_->SchedClosureAfter(micros, closure);
+  }
 
  private:
   Env* target_;