Home
last modified time | relevance | path

Searched refs:GetDistributedManager (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/eager/
H A Dremote_copy_node.cc112 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()
/aosp_15_r20/external/tensorflow/tensorflow/c/eager/
H A Dc_api_experimental.cc786 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()
H A Dimmediate_execution_context.h254 virtual ImmediateExecutionDistributedManager* GetDistributedManager() = 0;
H A Dc_api.cc199 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()
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/eager/
H A Dc_api_tfrt.h275 tensorflow::ImmediateExecutionDistributedManager* GetDistributedManager() in GetDistributedManager() function
280 return context_.GetEagerContext()->GetDistributedManager(); in GetDistributedManager()
H A Dc_api_tfrt.cc1488 if (context_->GetDistributedManager() != nullptr && in Initialize()
1491 context_->GetDistributedManager()) in Initialize()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/eager/
H A Dexecute.cc1849 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()
H A Dcontext.h506 ImmediateExecutionDistributedManager* GetDistributedManager() override { in GetDistributedManager() function