Home
last modified time | relevance | path

Searched refs:RdmaTensorResponse (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/contrib/verbs/
Drdma.h308 class RdmaTensorResponse {
311 RdmaTensorResponse(RdmaChannel* channel, const RdmaMessage& rm) in RdmaTensorResponse() function
374 friend class RdmaTensorResponse; variable
412 friend class RdmaTensorResponse; variable
439 RdmaTensorResponse* AddTensorResponse(const RdmaMessage& rm);
440 RdmaTensorResponse* UpdateTensorResponse(const RdmaMessage& rm);
469 typedef std::unordered_map<uint32_t, RdmaTensorResponse> ResponsesTable;
Drdma.cc488 RdmaTensorResponse* response = rc->AddTensorResponse(rm); in Process_CQ()
498 RdmaTensorResponse* response = rc->UpdateTensorResponse(rm); in Process_CQ()
518 RdmaTensorResponse* response = in Process_CQ()
519 reinterpret_cast<RdmaTensorResponse*>(wr_id->write_context); in Process_CQ()
972 RdmaTensorResponse* RdmaChannel::AddTensorResponse(const RdmaMessage& rm) { in AddTensorResponse()
975 responses_table_.emplace(rm.request_index_, RdmaTensorResponse(this, rm)); in AddTensorResponse()
981 RdmaTensorResponse* RdmaChannel::UpdateTensorResponse(const RdmaMessage& rm) { in UpdateTensorResponse()
985 RdmaTensorResponse* response = &it->second; in UpdateTensorResponse()
995 void RdmaTensorResponse::Start() { in Start()
1014 void RdmaTensorResponse::Resume() { SendContent(*tensor_, *proto_, is_dead_); } in Resume()
[all …]
DREADME.md64 …eceives an **RDMA_MESSAGE_TENSOR_REQUEST**, it will create a new **RdmaTensorResponse** object for…
82 … **RDMA_MESSAGE_TENSOR_RE_REQUEST**, it will locate the relevant **RdmaTensorResponse** using the …
98 …e. In the case where we need to send the tensor's meta-data, the **RdmaTensorResponse** will store…
120 * **class RdmaTensorResponse** - Holds and manages information for a single tensor response throu…
Dpatch_notes_verbs_with_0_copies.md44 * **class RdmaTensorResponse** - Holds information for a single tensor response, such as destinatio…