Home
last modified time | relevance | path

Searched refs:distributed_manager_ (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/tfrt/eager/
Dc_api_tfrt.h272 distributed_manager_ = std::move(distributed); in SetDistributedManager()
278 return distributed_manager_.get(); in GetDistributedManager()
319 distributed_manager_; variable
Dc_api_tfrt.cc957 return distributed_manager_->EnableCollectiveOps(server_def); in EnableCollectiveOps()
1018 if (distributed_manager_ != nullptr) { in BuildFunctionRequestContext()
1019 down_cast<DistributedManagerContextInterface*>(distributed_manager_.get()) in BuildFunctionRequestContext()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dcontext.h504 distributed_manager_ = std::move(distributed); in SetDistributedManager()
507 return distributed_manager_.get(); in GetDistributedManager()
818 std::unique_ptr<ImmediateExecutionDistributedManager> distributed_manager_; variable
Dcontext.cc1317 return distributed_manager_->EnableCollectiveOps(server_def); in EnableCollectiveOps()