Searched refs:GetDistributedManager (Results 1 – 8 of 8) sorted by relevance
112 if (ctx_->GetDistributedManager() != nullptr) in RunLocalSend()113 coord_agent = ctx_->GetDistributedManager()->GetCoordinationServiceAgent(); in RunLocalSend()203 if (ctx_->GetDistributedManager() != nullptr) in RunLocalRecv()204 coord_agent = ctx_->GetDistributedManager()->GetCoordinationServiceAgent(); in RunLocalRecv()
786 tensorflow::unwrap(ctx)->GetDistributedManager(); in TFE_InsertConfigKeyValue()800 tensorflow::unwrap(ctx)->GetDistributedManager(); in TFE_GetConfigKeyValue()825 tensorflow::unwrap(ctx)->GetDistributedManager(); in TFE_DeleteConfigKeyValue()839 tensorflow::unwrap(ctx)->GetDistributedManager(); in TFE_ReportErrorToCluster()
254 virtual ImmediateExecutionDistributedManager* GetDistributedManager() = 0;
199 tensorflow::unwrap(ctx)->GetDistributedManager()->SetOrUpdateServerDef( in TFE_ContextSetServerDef()226 tensorflow::unwrap(ctx)->GetDistributedManager()->SetOrUpdateServerDef( in TFE_ContextUpdateServerDef()241 tensorflow::unwrap(ctx)->GetDistributedManager()->CheckRemoteAlive( in TFE_ContextCheckAlive()
275 tensorflow::ImmediateExecutionDistributedManager* GetDistributedManager() in GetDistributedManager() function280 return context_.GetEagerContext()->GetDistributedManager(); in GetDistributedManager()
1488 if (context_->GetDistributedManager() != nullptr && in Initialize()1491 context_->GetDistributedManager()) in Initialize()
1849 if (ctx->GetDistributedManager() != nullptr) in EagerKernelExecute()1850 coord_agent = ctx->GetDistributedManager()->GetCoordinationServiceAgent(); in EagerKernelExecute()2057 if (ctx->GetDistributedManager() != nullptr) in EagerKernelExecuteAsync()2058 coord_agent = ctx->GetDistributedManager()->GetCoordinationServiceAgent(); in EagerKernelExecuteAsync()
506 ImmediateExecutionDistributedManager* GetDistributedManager() override { in GetDistributedManager() function