Searched refs:collective_executor_mgr (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | worker.cc | 356 if (env_->collective_executor_mgr) { in CleanupGraphAsync()
357 env_->collective_executor_mgr->Cleanup(step_id); in CleanupGraphAsync()
400 if (env_->collective_executor_mgr) { in CompleteGroupAsync()
401 env_->collective_executor_mgr->GetParamResolver()->CompleteGroupAsync( in CompleteGroupAsync()
413 if (env_->collective_executor_mgr) { in CompleteInstanceAsync()
414 env_->collective_executor_mgr->GetParamResolver()->CompleteInstanceAsync( in CompleteInstanceAsync()
425 if (env_->collective_executor_mgr) { in GetStepSequenceAsync()
426 env_->collective_executor_mgr->GetStepSequenceAsync(request, response, in GetStepSequenceAsync()
|
| D | worker_env.h | 63 std::unique_ptr<CollectiveExecutorMgrInterface> collective_executor_mgr; member
|
| D | master_env.h | 97 CollectiveExecutorMgrInterface* collective_executor_mgr = nullptr; member
|
| D | master_session.cc | 1575 uint64 step_id = env_->collective_executor_mgr->NextStepId(graph_key); in NewStepId()
1580 env_->collective_executor_mgr->RefreshStepIdSequenceAsync( in NewStepId()
1593 step_id = env_->collective_executor_mgr->NextStepId(graph_key); in NewStepId()
1913 env_->collective_executor_mgr->RetireStepId(rcg->collective_graph_key(), in PostRunCleanup()
|
| D | graph_mgr.cc | 463 worker_env_->collective_executor_mgr->FindOrCreate(step_id), in ExecuteAsync()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_server_lib.cc | 275 worker_env_.collective_executor_mgr.reset( in Init()
277 if (worker_env_.collective_executor_mgr == nullptr) { in Init()
288 worker_env_.collective_executor_mgr.reset(new RpcCollectiveExecutorMgr( in Init()
307 master_env_.collective_executor_mgr = in Init()
308 worker_env_.collective_executor_mgr.get(); in Init()
457 worker_env_.collective_executor_mgr.reset(new RpcCollectiveExecutorMgr( in UpdateServerDef()
463 master_env_.collective_executor_mgr = in UpdateServerDef()
464 worker_env_.collective_executor_mgr.get(); in UpdateServerDef()
|
| D | grpc_worker_service.cc | 635 env_->collective_executor_mgr->FindOrCreate(step_id), true); in RecvBufAsync()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | context.h | 293 CollectiveExecutorMgrInterface* collective_executor_mgr() { in collective_executor_mgr() function
299 collective_executor_mgr()->FindOrCreate(0), true /*inherit_ref*/)); in GetCollectiveExecutorHandle()
|
| D | context_distributed_manager.cc | 709 grpc_server->worker_env()->collective_executor_mgr.get())); in EnableCollectiveOps()
714 grpc_server->worker_env()->collective_executor_mgr.get())); in EnableCollectiveOps()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | BUILD | 218 "collective_executor_mgr.h",
332 name = "collective_executor_mgr",
333 srcs = ["collective_executor_mgr.cc"],
334 hdrs = ["collective_executor_mgr.h"],
1574 ":collective_executor_mgr",
|