Searched refs:BaseRemoteRendezvous (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | base_rendezvous_mgr.cc | 57 BaseRemoteRendezvous* BaseRendezvousMgr::FindOrCreate(int64 step_id) { in FindOrCreate()
133 BaseRemoteRendezvous::BaseRemoteRendezvous(const WorkerEnv* env, int64 step_id) in BaseRemoteRendezvous() function in tensorflow::BaseRemoteRendezvous
139 BaseRemoteRendezvous::~BaseRemoteRendezvous() { in ~BaseRemoteRendezvous()
152 Status BaseRemoteRendezvous::Initialize(WorkerSession* session) { in Initialize()
177 WorkerSession* BaseRemoteRendezvous::session() { in session()
182 bool BaseRemoteRendezvous::is_initialized() { in is_initialized()
187 Status BaseRemoteRendezvous::Send(const Rendezvous::ParsedKey& parsed, in Send()
205 Status BaseRemoteRendezvous::ValidateDevices(const ParsedKey& parsed, in ValidateDevices()
229 void BaseRemoteRendezvous::SameWorkerRecvDone( in SameWorkerRecvDone()
287 bool BaseRemoteRendezvous::IsSameWorker(DeviceNameUtils::ParsedName src, in IsSameWorker()
[all …]
|
| D | base_rendezvous_mgr.h | 39 class BaseRemoteRendezvous; variable
95 virtual BaseRemoteRendezvous* Create(int64 step_id,
100 typedef gtl::FlatMap<int64, BaseRemoteRendezvous*> Table;
108 BaseRemoteRendezvous* FindOrCreate(int64 step_id);
119 class BaseRemoteRendezvous : public RemoteRendezvous {
121 BaseRemoteRendezvous(const WorkerEnv* env, int64 step_id);
172 ~BaseRemoteRendezvous() override;
219 TF_DISALLOW_COPY_AND_ASSIGN(BaseRemoteRendezvous);
|
| /external/tensorflow/tensorflow/contrib/verbs/ |
| D | rdma_rendezvous_mgr.cc | 30 class RdmaRemoteRendezvous : public BaseRemoteRendezvous {
33 : BaseRemoteRendezvous(env, step_id), rdma_mgr_(rdma_mgr) {} in RdmaRemoteRendezvous()
85 BaseRemoteRendezvous* RdmaRendezvousMgr::Create(int64 step_id, in Create()
|
| D | rdma_rendezvous_mgr.h | 52 BaseRemoteRendezvous* Create(int64 step_id,
|
| D | README.md | 55 …!= sender) is to be handled in a derivation of the pure virtual **BaseRemoteRendezvous::RecvFromRe…
136 * **class RdmaRemoteRendezvous** - A derived class of BaseRemoteRendezvous. This class is the back …
|
| /external/tensorflow/tensorflow/contrib/gdr/ |
| D | gdr_rendezvous_mgr.cc | 121 class GdrRemoteRendezvous : public BaseRemoteRendezvous {
125 : BaseRemoteRendezvous(env, step_id), in GdrRemoteRendezvous()
211 BaseRemoteRendezvous* GdrRendezvousMgr::Create(int64 step_id, in Create()
|
| D | gdr_rendezvous_mgr.h | 32 BaseRemoteRendezvous* Create(int64 step_id, const WorkerEnv* worker_env);
|
| /external/tensorflow/tensorflow/contrib/mpi/ |
| D | mpi_rendezvous_mgr.h | 116 class MPIRemoteRendezvous : public BaseRemoteRendezvous {
120 : BaseRemoteRendezvous(env, step_id), in MPIRemoteRendezvous()
158 BaseRemoteRendezvous* Create(int64 step_id,
|
| D | mpi_rendezvous_mgr.cc | 57 BaseRemoteRendezvous* MPIRendezvousMgr::Create(int64 step_id, in Create()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | rpc_rendezvous_mgr.cc | 40 class RpcRemoteRendezvous : public BaseRemoteRendezvous {
43 : BaseRemoteRendezvous(env, step_id) {} in RpcRemoteRendezvous()
293 BaseRemoteRendezvous* RpcRendezvousMgr::Create(int64 step_id, in Create()
|
| D | rpc_rendezvous_mgr.h | 50 BaseRemoteRendezvous* Create(int64 step_id, const WorkerEnv* worker_env);
|