| /external/tensorflow/tensorflow/c/eager/ |
| D | immediate_execution_distributed_manager.h | 55 virtual CoordinationServiceAgent* GetCoordinationServiceAgent() = 0;
|
| D | c_api_experimental.cc | 788 dist_mgr->GetCoordinationServiceAgent(); in TFE_InsertConfigKeyValue()
802 dist_mgr->GetCoordinationServiceAgent(); in TFE_GetConfigKeyValue()
827 dist_mgr->GetCoordinationServiceAgent(); in TFE_DeleteConfigKeyValue()
841 dist_mgr->GetCoordinationServiceAgent(); in TFE_ReportErrorToCluster()
|
| /external/tensorflow/tensorflow/core/tfrt/eager/ |
| D | c_api_tfrt_distributed_impl.cc | 99 tensorflow::CoordinationServiceAgent* GetCoordinationServiceAgent() override;
192 DistributedManagerContextImpl::GetCoordinationServiceAgent() { in GetCoordinationServiceAgent() function in tfrt::tf::DistributedManagerContextImpl
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/distributed/ |
| D | client.cc | 62 GetCoordinationServiceAgent() override;
129 GetCoordinationServiceAgent() override;
380 DistributedRuntimeClientImpl::GetCoordinationServiceAgent() { in GetCoordinationServiceAgent() function in xla::DistributedRuntimeClientImpl
550 DistributedRuntimeCoordinationServiceClient::GetCoordinationServiceAgent() { in GetCoordinationServiceAgent() function in xla::DistributedRuntimeCoordinationServiceClient
|
| D | client.h | 128 GetCoordinationServiceAgent() = 0;
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | context_distributed_manager.h | 52 CoordinationServiceAgent* GetCoordinationServiceAgent() override { in GetCoordinationServiceAgent() function
|
| D | context_distributed_manager.cc | 610 session_mgr->GetCoordinationServiceAgent())); in UpdateContextWithServerDef()
676 ->session_mgr->GetCoordinationServiceAgent(); in SetOrUpdateServerDef()
722 coordination_service_agent_ = session_mgr->GetCoordinationServiceAgent(); in EnableCollectiveOps()
|
| D | execute.cc | 1850 coord_agent = ctx->GetDistributedManager()->GetCoordinationServiceAgent(); in EagerKernelExecute()
2058 coord_agent = ctx->GetDistributedManager()->GetCoordinationServiceAgent(); in EagerKernelExecuteAsync()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | session_mgr.h | 98 CoordinationServiceAgent* GetCoordinationServiceAgent();
|
| D | worker.cc | 229 collector, response, cm, env_->session_mgr->GetCoordinationServiceAgent(), in DoRunGraph()
322 env_->session_mgr->GetCoordinationServiceAgent(), in DoPartialRunGraph()
|
| D | session_mgr.cc | 336 CoordinationServiceAgent* SessionMgr::GetCoordinationServiceAgent() { in GetCoordinationServiceAgent() function in tensorflow::SessionMgr
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | remote_copy_node.cc | 113 coord_agent = ctx_->GetDistributedManager()->GetCoordinationServiceAgent(); in RunLocalSend()
204 coord_agent = ctx_->GetDistributedManager()->GetCoordinationServiceAgent(); in RunLocalRecv()
|
| D | eager_service_impl.cc | 321 auto coord_agent = env_->session_mgr->GetCoordinationServiceAgent(); in CreateContext()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/preemption/ |
| D | preemption_sync_manager.cc | 90 client->GetCoordinationServiceAgent()); in Initialize()
|