Searched refs:CollectiveExecutorMgr (Results 1 – 7 of 7) sorted by relevance
26 CollectiveExecutorMgr::CollectiveExecutorMgr( in CollectiveExecutorMgr() function in tensorflow::CollectiveExecutorMgr40 CollectiveExecutorMgr::~CollectiveExecutorMgr() { in ~CollectiveExecutorMgr()46 CollectiveExecutor* CollectiveExecutorMgr::FindOrCreate(int64 step_id) { in FindOrCreate()62 CollectiveExecutor* CollectiveExecutorMgr::Create(int64 step_id) { in Create()69 void CollectiveExecutorMgr::Cleanup(int64 step_id) { in Cleanup()82 void CollectiveExecutorMgr::GetStepSequenceAsync( in GetStepSequenceAsync()89 void CollectiveExecutorMgr::RefreshStepIdSequenceAsync( in RefreshStepIdSequenceAsync()
27 class CollectiveExecutorMgr : public CollectiveExecutorMgrInterface {29 CollectiveExecutorMgr(35 virtual ~CollectiveExecutorMgr();
50 cme_.reset(new CollectiveExecutorMgr(cp, device_mgr_.get(), std::move(drl), in CollectiveExecutorMgrTest()55 std::unique_ptr<CollectiveExecutorMgr> cme_;
557 collective_executor_mgr_.reset(new CollectiveExecutorMgr( in RunInternal()
35 class RpcCollectiveExecutorMgr : public CollectiveExecutorMgr {
34 : CollectiveExecutorMgr(config, dev_mgr, std::move(dev_resolver), in RpcCollectiveExecutorMgr()
125 new CollectiveExecutorMgr(opts.config, local_device_mgr(), std::move(drl),