Searched refs:RdmaTensorRequest (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/contrib/verbs/ |
| D | rdma.h | 234 class RdmaTensorRequest {
239 RdmaTensorRequest(uint32_t index, const string& key, int64 step_id,
243 ~RdmaTensorRequest();
411 friend class RdmaTensorRequest; variable
431 RdmaTensorRequest* InsertTensorRequest(
434 const RdmaTensorRequest::RecvDoneCallback& done);
436 RdmaTensorRequest* GetTensorRequest(uint32_t request_index);
465 typedef std::unordered_map<uint32_t, RdmaTensorRequest> RequestTable;
|
| D | rdma.cc | 474 RdmaTensorRequest* request = rc->GetTensorRequest(request_index); in Process_CQ()
491 RdmaTensorRequest* request = rc->GetTensorRequest(rm.request_index_); in Process_CQ()
501 RdmaTensorRequest* request = rc->GetTensorRequest(rm.request_index_); in Process_CQ()
667 RdmaTensorRequest* RdmaChannel::InsertTensorRequest( in InsertTensorRequest()
670 const RdmaTensorRequest::RecvDoneCallback& done) { in InsertTensorRequest()
676 RdmaTensorRequest request(request_index, key, step_id, this, dst_dev, in InsertTensorRequest()
687 RdmaTensorRequest* RdmaChannel::GetTensorRequest(uint32_t request_index) { in GetTensorRequest()
1476 RdmaTensorRequest::RdmaTensorRequest( in RdmaTensorRequest() function in tensorflow::RdmaTensorRequest
1479 const RdmaTensorRequest::RecvDoneCallback& done) in RdmaTensorRequest()
1493 RdmaTensorRequest::~RdmaTensorRequest() { DeallocateTensors(); } in ~RdmaTensorRequest()
[all …]
|
| D | rdma_rendezvous_mgr.cc | 77 RdmaTensorRequest* request = in RecvFromRemoteAsync()
|
| D | README.md | 57 …cated in rdma_rendezvous_mgr.cc. The implementation creates a new **RdmaTensorRequest** object, ke…
76 … **RDMA_MESSAGE_META_DATA_RESPONSE**, it will locate the relevant **RdmaTensorRequest** using the …
84 When the receiver receives the RDMA write, it will locate the relevant **RdmaTensorRequest** using …
109 * **class RdmaTensorRequest** - Holds and manages information for a single tensor request throug…
|
| D | patch_notes_verbs_with_0_copies.md | 40 * **class RdmaTensorRequest** - Holds and manages information for a single tensor request througho…
|