MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / LookupAndPin

Method LookupAndPin

tensorflow/compiler/xrt/xrt_memory_manager.cc:189–196  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

187}
188
189Status XRTMemoryManager::WorkingSet::LookupAndPin(xla::Backend* backend,
190 int64 handle) {
191 TF_ASSIGN_OR_RETURN(auto tuple, memory_manager_->Lookup(handle));
192 TF_RETURN_IF_ERROR(
193 tuple->PinAndSwapIn(memory_manager_.get(), backend).status());
194 pinned_tuples_.push_back(std::move(tuple));
195 return Status::OK();
196}
197
198/* static */ RefPtr<XRTMemoryManager> XRTMemoryManager::Get(ResourceMgr* rm) {
199 static string* container = new string("XrtState");

Callers 2

PopulateOpWorkingSetFunction · 0.80
GetInputBuffersFunction · 0.80

Calls 4

PinAndSwapInMethod · 0.80
statusMethod · 0.45
getMethod · 0.45
push_backMethod · 0.45

Tested by

no test coverage detected