Searched refs:RpcCollectiveExecutorMgr (Results 1 – 4 of 4) sorted by relevance
28 RpcCollectiveExecutorMgr::RpcCollectiveExecutorMgr( in RpcCollectiveExecutorMgr() function in tensorflow::RpcCollectiveExecutorMgr44 RpcCollectiveExecutorMgr::~RpcCollectiveExecutorMgr() { in ~RpcCollectiveExecutorMgr()50 CollectiveExecutor* RpcCollectiveExecutorMgr::Create(int64 step_id) { in Create()71 void RpcCollectiveExecutorMgr::RefreshStepIdSequenceAsync( in RefreshStepIdSequenceAsync()106 void RpcCollectiveExecutorMgr::GetStepSequenceAsync( in GetStepSequenceAsync()132 Status RpcCollectiveExecutorMgr::UpdateStepSequences( in UpdateStepSequences()149 int64 RpcCollectiveExecutorMgr::NextStepId(int64 graph_key) { in NextStepId()158 void RpcCollectiveExecutorMgr::RetireStepId(int64 graph_key, int64 step_id) { in RetireStepId()
35 class RpcCollectiveExecutorMgr : public CollectiveExecutorMgr {37 RpcCollectiveExecutorMgr(44 virtual ~RpcCollectiveExecutorMgr();
58 cme_.reset(new RpcCollectiveExecutorMgr( in RpcCollectiveExecutorMgrTest()63 std::unique_ptr<RpcCollectiveExecutorMgr> cme_;
288 worker_env_.collective_executor_mgr.reset(new RpcCollectiveExecutorMgr( in Init()457 worker_env_.collective_executor_mgr.reset(new RpcCollectiveExecutorMgr( in UpdateServerDef()