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