diff options
Diffstat (limited to 'tensorflow/core/distributed_runtime/master.h')
| -rw-r--r-- | tensorflow/core/distributed_runtime/master.h | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/tensorflow/core/distributed_runtime/master.h b/tensorflow/core/distributed_runtime/master.h index ce05a6508b..678fc46bd7 100644 --- a/tensorflow/core/distributed_runtime/master.h +++ b/tensorflow/core/distributed_runtime/master.h @@ -94,6 +94,10 @@ class Master { // Cleanup unused session. void GC(); + // Find master session by session handle, and increments the reference count + // on the returned MasterSession if not null. + MasterSession* FindMasterSession(const string& handle); + TF_DISALLOW_COPY_AND_ASSIGN(Master); }; |