| 187 | } |
| 188 | |
| 189 | Status 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"); |
no test coverage detected