Home
last modified time | relevance | path

Searched refs:RdmaMemoryMgr (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/contrib/verbs/
Drdma_mgr.cc262 flag, [this]() { RdmaMemoryMgr::Singleton().pd_ = rdma_adapter_->pd_; }); in InitAllocators()
268 RdmaMemoryMgr::Singleton().InsertMemoryRegion( in RegMemVisitors()
273 RdmaMemoryMgr::Singleton().EvictMemoryRegion(ptr, num_bytes); in RegMemVisitors()
295 RdmaMemoryMgr::Singleton().InsertMemoryRegion( in RegMemVisitors()
Drdma.cc1077 (RdmaMemoryMgr::Singleton().FindMemoryRegion( in RecvHandler()
1207 mr_ = RdmaMemoryMgr::Singleton().FindMemoryRegion(src_addr_, in SendContent()
1410 ibv_mr* RdmaMemoryMgr::FindMemoryRegion(void* addr, size_t length) { in FindMemoryRegion()
1420 void RdmaMemoryMgr::InsertMemoryRegion(void* addr, size_t length, in InsertMemoryRegion()
1437 void RdmaMemoryMgr::EvictMemoryRegion(void* addr, size_t length) { in EvictMemoryRegion()
1450 const TensorMetaData* RdmaMemoryMgr::GetTensorMetaData( in GetTensorMetaData()
1460 const TensorMetaData* RdmaMemoryMgr::SetTensorMetaData( in SetTensorMetaData()
1486 meta_data_(RdmaMemoryMgr::Singleton().GetTensorMetaData(key)), in RdmaTensorRequest()
1541 mr_ = RdmaMemoryMgr::Singleton().FindMemoryRegion(rdma_addr_, tensor_size); in AllocateTensors()
1550 RdmaMemoryMgr::Singleton().FindMemoryRegion(rdma_addr_, tensor_size); in AllocateTensors()
[all …]
Drdma.h194 class RdmaMemoryMgr {
196 static RdmaMemoryMgr& Singleton() { in Singleton()
197 static RdmaMemoryMgr instance; in Singleton()
217 RdmaMemoryMgr() : pd_(nullptr) {} in RdmaMemoryMgr() function
Dpatch_notes_verbs_with_0_copies.md39 * **class RdmaMemoryMgr** - Manages the meta-data cache, and the registered memory regions.
DREADME.md108 * **class RdmaMemoryMgr** - Manages the meta-data cache, and the registered memory regions.