Home | Sort by relevance Sort by last modified time |
/external/tensorflow/tensorflow/contrib/verbs/ | |
rdma_rendezvous_mgr.h | 49 void SetRdmaMgr(RdmaMgr* rdma_mgr) { rdma_mgr_ = rdma_mgr; } 56 RdmaMgr* rdma_mgr_; member in class:tensorflow::RdmaRendezvousMgr |
grpc_verbs_service.h | 42 void SetRdmaMgr(RdmaMgr* rdma_mgr) { rdma_mgr_ = rdma_mgr; } 59 RdmaMgr* rdma_mgr_; member in class:tensorflow::GrpcVerbsService |
rdma_rendezvous_mgr.cc | 33 : BaseRemoteRendezvous(env, step_id), rdma_mgr_(rdma_mgr) {} 42 RdmaMgr* rdma_mgr_; member in class:tensorflow::RdmaRemoteRendezvous 64 CHECK(dst_name.compare(rdma_mgr_->local_worker()) == 0); 65 RdmaChannel* rc = rdma_mgr_->FindChannel(src_name); 87 return new RdmaRemoteRendezvous(worker_env, step_id, rdma_mgr_); |
verbs_server_lib.h | 50 RdmaMgr* rdma_mgr_; member in class:tensorflow::VerbsServer |