Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/runtime_fallback/runtime/
Dkernel_utils.cc49 owned_eager_context->SetDistributedManager( in InitEagerContext()
66 owned_eager_context->SetDistributedManager( in InitEagerContext()
/external/tensorflow/tensorflow/c/eager/
Dimmediate_execution_context.h242 virtual void SetDistributedManager(
Dc_api.cc124 tfrt_context->SetDistributedManager( in TFE_NewContext()
154 eager_context->SetDistributedManager( in TFE_NewContext()
/external/tensorflow/tensorflow/core/tfrt/eager/
Dc_api_tfrt.h251 void SetDistributedManager( in SetDistributedManager() function
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dcontext.h465 void SetDistributedManager( in SetDistributedManager() function
/external/tensorflow/tensorflow/core/distributed_runtime/eager/
Deager_service_impl.cc316 ctx->SetDistributedManager(std::move(dist_mgr)); in CreateContext()