Searched refs:CollectiveExecutorMgr (Results 1 – 7 of 7) sorted by relevance
26 CollectiveExecutorMgr::CollectiveExecutorMgr( in CollectiveExecutorMgr() function in tensorflow::CollectiveExecutorMgr36 CollectiveExecutorMgr::~CollectiveExecutorMgr() { in ~CollectiveExecutorMgr()42 CollectiveExecutor* CollectiveExecutorMgr::FindOrCreate(int64 step_id) { in FindOrCreate()58 CollectiveExecutor* CollectiveExecutorMgr::Create(int64 step_id) { in Create()65 void CollectiveExecutorMgr::Cleanup(int64 step_id) { in Cleanup()78 void CollectiveExecutorMgr::GetStepSequenceAsync( in GetStepSequenceAsync()85 void CollectiveExecutorMgr::RefreshStepIdSequenceAsync( in RefreshStepIdSequenceAsync()
25 class CollectiveExecutorMgr : public CollectiveExecutorMgrInterface {27 CollectiveExecutorMgr(const ConfigProto& config, const DeviceMgr* dev_mgr,31 virtual ~CollectiveExecutorMgr();
49 cme_.reset(new CollectiveExecutorMgr(cp, device_mgr_.get(), std::move(drl), in CollectiveExecutorMgrTest()53 std::unique_ptr<CollectiveExecutorMgr> cme_;
507 collective_executor_mgr_.reset(new CollectiveExecutorMgr( in RunInternal()
35 class RpcCollectiveExecutorMgr : public CollectiveExecutorMgr {
33 : CollectiveExecutorMgr(config, dev_mgr, std::move(dev_resolver), in RpcCollectiveExecutorMgr()
89 collective_executor_mgr_.reset(new CollectiveExecutorMgr( in EagerContext()