Home
last modified time | relevance | path

Searched refs:RdmaChannel (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/contrib/verbs/
Drdma.h187 class RdmaChannel; variable
240 RdmaChannel* channel, Device* dst_dev,
292 RdmaChannel* channel_;
311 RdmaTensorResponse(RdmaChannel* channel, const RdmaMessage& rm) in RdmaTensorResponse()
352 RdmaChannel* channel_;
372 friend class RdmaChannel; variable
407 class RdmaChannel {
417 explicit RdmaChannel(const RdmaAdapter* adapter, const string local_name,
419 ~RdmaChannel();
478 friend class RdmaChannel; variable
[all …]
Drdma_mgr.cc52 new RdmaChannel(rdma_adapter_, local_worker_, workers[i])}); in RdmaMgr()
64 RdmaChannel* rc = p.second; in SetupChannels()
81 for (int i = 0; i < RdmaChannel::kNumMessageBuffers; i++) { in SetupChannels()
119 CHECK(i == RdmaChannel::kNumMessageBuffers); in SetupChannels()
141 RdmaChannel* rc = p.second; in ConnectivityCheck()
163 if ((int)rdma_adapter_->wc_[i].wr_id == RdmaChannel::kPingRecvWrid) { in ConnectivityCheck()
170 RdmaChannel* rc = in ConnectivityCheck()
171 reinterpret_cast<RdmaChannel*>(rdma_adapter_->wc_[i].wr_id); in ConnectivityCheck()
195 RdmaChannel* RdmaMgr::FindChannel(const string& name) { in FindChannel()
Drdma_mgr.h31 friend class RdmaChannel; variable
38 RdmaChannel* FindChannel(const string& key);
51 typedef std::unordered_map<string, RdmaChannel*> ChannelTable;
Dgrpc_verbs_service.cc114 RdmaChannel* rc = rdma_mgr_->FindChannel(remote_host_name); in GetRemoteAddressSync()
127 CHECK_EQ(request->mr_size(), RdmaChannel::kNumMessageBuffers); in GetRemoteAddressSync()
140 CHECK(i == RdmaChannel::kNumMessageBuffers); in GetRemoteAddressSync()
151 for (int i = 0; i < RdmaChannel::kNumMessageBuffers; i++) { in GetRemoteAddressSync()
Drdma.cc455 RdmaChannel* rc = reinterpret_cast<RdmaChannel*>(wc_[i].wr_id); in Process_CQ()
529 int RdmaChannel::PingPostRecv() { in PingPostRecv()
539 int RdmaChannel::PingPostSend() { in PingPostSend()
551 RdmaChannel::RdmaChannel(const RdmaAdapter* adapter, const string local_name, in RdmaChannel() function in tensorflow::RdmaChannel
632 RdmaChannel::~RdmaChannel() { in ~RdmaChannel()
639 void RdmaChannel::SetRemoteAddress(const RdmaAddress& ra, bool override) { in SetRemoteAddress()
659 void RdmaChannel::Recv() { in Recv()
667 RdmaTensorRequest* RdmaChannel::InsertTensorRequest( in InsertTensorRequest()
682 void RdmaChannel::RemoveTensorRequest(uint32_t request_index) { in RemoveTensorRequest()
687 RdmaTensorRequest* RdmaChannel::GetTensorRequest(uint32_t request_index) { in GetTensorRequest()
[all …]
Drdma_rendezvous_mgr.cc65 RdmaChannel* rc = rdma_mgr_->FindChannel(src_name); in RecvFromRemoteAsync()
DREADME.md132 * **class RdmaChannel** - Responsible for RDMA connection to a particular node. It manages…