Searched refs:collective_executor_mgr (Results 1 – 11 of 11) sorted by relevance
351 if (env_->collective_executor_mgr) { in CleanupGraphAsync()352 env_->collective_executor_mgr->Cleanup(step_id); in CleanupGraphAsync()395 if (env_->collective_executor_mgr) { in CompleteGroupAsync()396 env_->collective_executor_mgr->GetParamResolver()->CompleteGroupAsync( in CompleteGroupAsync()408 if (env_->collective_executor_mgr) { in CompleteInstanceAsync()409 env_->collective_executor_mgr->GetParamResolver()->CompleteInstanceAsync( in CompleteInstanceAsync()420 if (env_->collective_executor_mgr) { in GetStepSequenceAsync()421 env_->collective_executor_mgr->GetStepSequenceAsync(request, response, in GetStepSequenceAsync()
63 CollectiveExecutorMgrInterface* collective_executor_mgr = nullptr; member
97 CollectiveExecutorMgrInterface* collective_executor_mgr = nullptr; member
1483 uint64 step_id = env_->collective_executor_mgr->NextStepId(graph_key); in NewStepId()1488 env_->collective_executor_mgr->RefreshStepIdSequenceAsync( in NewStepId()1501 step_id = env_->collective_executor_mgr->NextStepId(graph_key); in NewStepId()1821 env_->collective_executor_mgr->RetireStepId(rcg->collective_graph_key(), in PostRunCleanup()
431 worker_env_->collective_executor_mgr->FindOrCreate(step_id), in ExecuteAsync()
222 worker_env_.collective_executor_mgr = in Init()224 if (!worker_env_.collective_executor_mgr) { in Init()236 worker_env_.collective_executor_mgr = new RpcCollectiveExecutorMgr( in Init()254 master_env_.collective_executor_mgr = worker_env_.collective_executor_mgr; in Init()
565 env_->collective_executor_mgr->FindOrCreate(request->step_id()), true); in RecvBufAsync()
161 CollectiveExecutorMgrInterface* collective_executor_mgr() { in collective_executor_mgr() function169 collective_executor_mgr()->FindOrCreate(0), true /*inherit_ref*/)); in GetCollectiveExecutorHandle()
160 env_->collective_executor_mgr->FindOrCreate(request->step_id()), true); in RecvBufAsync()
3035 "common_runtime/collective_executor_mgr.h",3088 "common_runtime/collective_executor_mgr.cc",
8944 grpc_server->worker_env()->collective_executor_mgr)); in EnableCollectiveOps()