diff options
Diffstat (limited to 'tensorflow/core/distributed_runtime/rpc_collective_executor_mgr.h')
-rw-r--r-- | tensorflow/core/distributed_runtime/rpc_collective_executor_mgr.h | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/tensorflow/core/distributed_runtime/rpc_collective_executor_mgr.h b/tensorflow/core/distributed_runtime/rpc_collective_executor_mgr.h index e9f3f0ebe8..c9581fa00f 100644 --- a/tensorflow/core/distributed_runtime/rpc_collective_executor_mgr.h +++ b/tensorflow/core/distributed_runtime/rpc_collective_executor_mgr.h @@ -42,6 +42,12 @@ class RpcCollectiveExecutorMgr : public CollectiveExecutorMgr { virtual ~RpcCollectiveExecutorMgr(); + // This function should only be called at the group_leader, by an RPC. + // Other needs for StepIds should be satisfied by NextStepId. + void GetStepSequenceAsync(const GetStepSequenceRequest* request, + GetStepSequenceResponse* response, + const StatusCallback& done) override; + void RefreshStepIdSequenceAsync(int64 graph_key, const StatusCallback& done) override; |