Searched defs:MetadataIndex (Results 1 – 5 of 5) sorted by relevance
283 def _find_shard(tensor: ShardedTensor, index: MetadataIndex) -> Shard:304 def find_tensor_shard(tensor: torch.Tensor, index: MetadataIndex) -> torch.Tensor:320 def find_state_dict_object(state_dict: STATE_DICT_TYPE, index: MetadataIndex) -> Any:
210 def lookup_tensor(self, index: MetadataIndex) -> torch.Tensor:
140 def lookup_object(self, index: MetadataIndex) -> Any:262 def lookup_tensor(self, index: MetadataIndex) -> torch.Tensor:
151 class MetadataIndex: class
278 def __get_tensor_shard__(self, index: MetadataIndex) -> torch.Tensor: