Home
last modified time | relevance | path

Searched refs:BaseRendezvousMgr (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/core/distributed_runtime/
Dbase_rendezvous_mgr.cc43 BaseRendezvousMgr::BaseRendezvousMgr(const WorkerEnv* worker_env) in BaseRendezvousMgr() function in tensorflow::BaseRendezvousMgr
46 BaseRendezvousMgr::~BaseRendezvousMgr() { in ~BaseRendezvousMgr()
53 RemoteRendezvous* BaseRendezvousMgr::Find(int64 step_id) { in Find()
57 BaseRemoteRendezvous* BaseRendezvousMgr::FindOrCreate(int64 step_id) { in FindOrCreate()
68 void BaseRendezvousMgr::RecvLocalAsync(int64 step_id, in RecvLocalAsync()
85 Status BaseRendezvousMgr::RecvLocal(int64 step_id, in RecvLocal()
104 void BaseRendezvousMgr::Cleanup(int64 step_id) { in Cleanup()
119 void BaseRendezvousMgr::CleanupAll() { in CleanupAll()
Dbase_rendezvous_mgr.h60 class BaseRendezvousMgr : public RendezvousMgrInterface {
62 explicit BaseRendezvousMgr(const WorkerEnv* worker_env);
64 ~BaseRendezvousMgr() override;
110 TF_DISALLOW_COPY_AND_ASSIGN(BaseRendezvousMgr);
/external/tensorflow/tensorflow/contrib/mpi/
Dmpi_rendezvous_mgr.h119 BaseRendezvousMgr* mgr_) in MPIRemoteRendezvous()
133 BaseRendezvousMgr* rendezvous_mgr_;
138 class MPIRendezvousMgr : public BaseRendezvousMgr {
Dmpi_rendezvous_mgr.cc36 : BaseRendezvousMgr(env), in MPIRendezvousMgr()
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Drpc_rendezvous_mgr.h45 class RpcRendezvousMgr : public BaseRendezvousMgr {
Drpc_rendezvous_mgr.cc291 : BaseRendezvousMgr(env) {} in RpcRendezvousMgr()
/external/tensorflow/tensorflow/contrib/gdr/
Dgdr_rendezvous_mgr.h26 class GdrRendezvousMgr : public BaseRendezvousMgr {
Dgdr_rendezvous_mgr.cc209 : BaseRendezvousMgr(env), remote_memory_manager_(remote_memory_manager) {} in GdrRendezvousMgr()
/external/tensorflow/tensorflow/contrib/verbs/
Drdma_rendezvous_mgr.h46 class RdmaRendezvousMgr : public BaseRendezvousMgr {
Drdma_rendezvous_mgr.cc83 : BaseRendezvousMgr(env) {} in RdmaRendezvousMgr()