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