OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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