Searched defs:resolve_tensor (Results 1 – 2 of 2) sorted by relevance
| /external/pytorch/torch/distributed/checkpoint/ | ||
| D | planner.py | 398 def resolve_tensor(self, read_item: ReadItem) -> torch.Tensor: member in LoadPlanner |
| D | default_planner.py | 255 def resolve_tensor(self, read_item: ReadItem): member in DefaultLoadPlanner |