Home
last modified time | relevance | path

Searched refs:SetDistributedManager (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/runtime/
H A Dkernel_utils.cc49 owned_eager_context->SetDistributedManager( in InitEagerContext()
66 owned_eager_context->SetDistributedManager( in InitEagerContext()
/aosp_15_r20/external/tensorflow/tensorflow/c/eager/
H A Dimmediate_execution_context.h251 virtual void SetDistributedManager(
H A Dc_api.cc125 tfrt_context->SetDistributedManager( in TFE_NewContext()
156 eager_context->SetDistributedManager( in TFE_NewContext()
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/eager/
H A Dc_api_tfrt.h269 void SetDistributedManager( in SetDistributedManager() function
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/eager/
H A Dcontext.h501 void SetDistributedManager( in SetDistributedManager() function
/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/eager/
H A Deager_service_impl.cc339 ctx->SetDistributedManager(std::move(dist_mgr)); in CreateContext()