HomeSort by relevance Sort by last modified time
    Searched refs:BaseRemoteRendezvous (Results 1 - 11 of 11) sorted by null

  /external/tensorflow/tensorflow/core/distributed_runtime/rpc/
rpc_rendezvous_mgr.h 50 BaseRemoteRendezvous* Create(int64 step_id, const WorkerEnv* worker_env);
rpc_rendezvous_mgr.cc 40 class RpcRemoteRendezvous : public BaseRemoteRendezvous {
43 : BaseRemoteRendezvous(env, step_id) {}
268 BaseRemoteRendezvous* RpcRendezvousMgr::Create(int64 step_id,
  /external/tensorflow/tensorflow/core/distributed_runtime/
base_rendezvous_mgr.h 39 class BaseRemoteRendezvous;
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);
123 // Upgrades the BaseRemoteRendezvous to full initialization.
172 ~BaseRemoteRendezvous() override;
219 TF_DISALLOW_COPY_AND_ASSIGN(BaseRemoteRendezvous);
base_rendezvous_mgr.cc 57 BaseRemoteRendezvous* BaseRendezvousMgr::FindOrCreate(int64 step_id) {
133 BaseRemoteRendezvous::BaseRemoteRendezvous(const WorkerEnv* env, int64 step_id)
139 BaseRemoteRendezvous::~BaseRemoteRendezvous() {
152 Status BaseRemoteRendezvous::Initialize(WorkerSession* session) {
177 WorkerSession* BaseRemoteRendezvous::session() {
182 bool BaseRemoteRendezvous::is_initialized() {
187 Status BaseRemoteRendezvous::Send(const Rendezvous::ParsedKey& parsed,
190 VLOG(1) << "BaseRemoteRendezvous Send " << this << " " << parsed.FullKey()
    [all...]
  /external/tensorflow/tensorflow/contrib/gdr/
gdr_rendezvous_mgr.h 32 BaseRemoteRendezvous* Create(int64 step_id, const WorkerEnv* worker_env);
gdr_rendezvous_mgr.cc 126 class GdrRemoteRendezvous : public BaseRemoteRendezvous {
130 : BaseRemoteRendezvous(env, step_id),
202 BaseRemoteRendezvous* GdrRendezvousMgr::Create(int64 step_id,
  /external/tensorflow/tensorflow/contrib/verbs/
rdma_rendezvous_mgr.h 52 BaseRemoteRendezvous* Create(int64 step_id,
rdma_rendezvous_mgr.cc 30 class RdmaRemoteRendezvous : public BaseRemoteRendezvous {
33 : BaseRemoteRendezvous(env, step_id), rdma_mgr_(rdma_mgr) {}
85 BaseRemoteRendezvous* RdmaRendezvousMgr::Create(int64 step_id,
README.md 55 The tensor transfer process is initiated when the receiver requests a tensor. In code it is done by calling **Rendezvous::Recv()** or **Rendezvous::RecvAsync()**. The TensorFlow base implementation handles the case where the requested tensor is located on the same node. The more interesting case where the requested tensor is located on a remote node (receiver != sender) is to be handled in a derivation of the pure virtual **BaseRemoteRendezvous::RecvFromRemoteAsync()**. TensorFlow provides a default GRPC based implementation which comes in the vanilla version but suffers in scalability when running large models. Our RDMA based implementation presumes to be more scalable. HKUST's contrib GDR implementation is more scalable than GRPC, and less scalable than ours, only because we did our evolution based on it.
136 * **class RdmaRemoteRendezvous** - A derived class of BaseRemoteRendezvous. This class is the back end for "send" and "recv" ops. When the sendrecv_op wants to send or receive a tensor, it calls the rendezvous' "send" and "recv" functions respectively. Rendezvous are identified by "step_id", a random number, so that tensors for different iterations don't get mixed up.
  /external/tensorflow/tensorflow/contrib/mpi/
mpi_rendezvous_mgr.h 116 class MPIRemoteRendezvous : public BaseRemoteRendezvous {
120 : BaseRemoteRendezvous(env, step_id),
158 BaseRemoteRendezvous* Create(int64 step_id,
mpi_rendezvous_mgr.cc 57 BaseRemoteRendezvous* MPIRendezvousMgr::Create(int64 step_id,

Completed in 227 milliseconds