Searched refs:RpcCollectiveExecutorMgr (Results 1 – 5 of 5) sorted by relevance
28 RpcCollectiveExecutorMgr::RpcCollectiveExecutorMgr( in RpcCollectiveExecutorMgr() function in tensorflow::RpcCollectiveExecutorMgr42 RpcCollectiveExecutorMgr::~RpcCollectiveExecutorMgr() { in ~RpcCollectiveExecutorMgr()48 CollectiveExecutor* RpcCollectiveExecutorMgr::Create(int64 step_id) { in Create()68 void RpcCollectiveExecutorMgr::RefreshStepIdSequenceAsync( in RefreshStepIdSequenceAsync()103 void RpcCollectiveExecutorMgr::GetStepSequenceAsync( in GetStepSequenceAsync()129 Status RpcCollectiveExecutorMgr::UpdateStepSequences( in UpdateStepSequences()146 int64 RpcCollectiveExecutorMgr::NextStepId(int64 graph_key) { in NextStepId()155 void RpcCollectiveExecutorMgr::RetireStepId(int64 graph_key, int64 step_id) { in RetireStepId()
35 class RpcCollectiveExecutorMgr : public CollectiveExecutorMgr {37 RpcCollectiveExecutorMgr(43 virtual ~RpcCollectiveExecutorMgr();
55 cme_.reset(new RpcCollectiveExecutorMgr(options.config, device_mgr_.get(), in RpcCollectiveExecutorMgrTest()60 std::unique_ptr<RpcCollectiveExecutorMgr> cme_;
33 class GdrCollectiveExecutorMgr : public RpcCollectiveExecutorMgr {41 : RpcCollectiveExecutorMgr(config, dev_mgr, std::move(dev_resolver), in GdrCollectiveExecutorMgr()
236 worker_env_.collective_executor_mgr = new RpcCollectiveExecutorMgr( in Init()