Searched refs:collective_executor_mgr (Results 1 – 10 of 10) sorted by relevance
353 if (env_->collective_executor_mgr) { in CleanupGraphAsync()354 env_->collective_executor_mgr->Cleanup(step_id); in CleanupGraphAsync()397 if (env_->collective_executor_mgr) { in CompleteGroupAsync()398 env_->collective_executor_mgr->GetParamResolver()->CompleteGroupAsync( in CompleteGroupAsync()410 if (env_->collective_executor_mgr) { in CompleteInstanceAsync()411 env_->collective_executor_mgr->GetParamResolver()->CompleteInstanceAsync( in CompleteInstanceAsync()422 if (env_->collective_executor_mgr) { in GetStepSequenceAsync()423 env_->collective_executor_mgr->GetStepSequenceAsync(request, response, in GetStepSequenceAsync()
63 CollectiveExecutorMgrInterface* collective_executor_mgr = nullptr; member
97 CollectiveExecutorMgrInterface* collective_executor_mgr = nullptr; member
1553 uint64 step_id = env_->collective_executor_mgr->NextStepId(graph_key); in NewStepId()1558 env_->collective_executor_mgr->RefreshStepIdSequenceAsync( in NewStepId()1571 step_id = env_->collective_executor_mgr->NextStepId(graph_key); in NewStepId()1891 env_->collective_executor_mgr->RetireStepId(rcg->collective_graph_key(), in PostRunCleanup()
451 worker_env_->collective_executor_mgr->FindOrCreate(step_id), in ExecuteAsync()
255 worker_env_.collective_executor_mgr = in Init()257 if (!worker_env_.collective_executor_mgr) { in Init()269 worker_env_.collective_executor_mgr = new RpcCollectiveExecutorMgr( in Init()287 master_env_.collective_executor_mgr = worker_env_.collective_executor_mgr; in Init()435 worker_env_.collective_executor_mgr = new RpcCollectiveExecutorMgr( in UpdateServerDef()441 master_env_.collective_executor_mgr = worker_env_.collective_executor_mgr; in UpdateServerDef()
635 env_->collective_executor_mgr->FindOrCreate(step_id), true); in RecvBufAsync()
248 CollectiveExecutorMgrInterface* collective_executor_mgr() { in collective_executor_mgr() function254 collective_executor_mgr()->FindOrCreate(0), true /*inherit_ref*/)); in GetCollectiveExecutorHandle()
786 grpc_server->worker_env()->collective_executor_mgr)); in EnableCollectiveOps()791 grpc_server->worker_env()->collective_executor_mgr)); in EnableCollectiveOps()
2452 "common_runtime/collective_executor_mgr.h",2508 "common_runtime/collective_executor_mgr.cc",