(self, data)
| 295 | """ |
| 296 | |
| 297 | def _load_cache(self, data): |
| 298 | if isinstance(data, Tensor): |
| 299 | if data.device == self.default_device: |
| 300 | return data |
| 301 | return copy(data, device=self.default_device) |
| 302 | elif isinstance(data, collections.abc.Mapping): |
| 303 | return {k: self._load_cache(v) for k, v in data.items()} |
| 304 | elif isinstance(data, tuple) and hasattr(data, "_fields"): # namedtuple |
| 305 | return type(data)(*(self._load_cache(value) for value in data)) |
| 306 | elif isinstance(data, collections.abc.Sequence): |
| 307 | return [self._load_cache(value) for value in data] |
| 308 | else: |
| 309 | return data |
| 310 | |
| 311 | def _swap_out_cache(self): |
| 312 | out = self._load_cache(self.pre_load_device_cache) |
no test coverage detected