Searched defs:lookup_tensor (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/distributed/checkpoint/ | ||
| H A D | optimizer.py | 210 def lookup_tensor(self, index: MetadataIndex) -> torch.Tensor: member in _ReaderWithOffset |
| H A D | default_planner.py | 262 def lookup_tensor(self, index: MetadataIndex) -> torch.Tensor: member in DefaultLoadPlanner |