Home
last modified time | relevance | path

Searched defs:load_tensor (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/test/distributed/checkpoint/
H A Dtest_file_system_checkpoint.py218 def load_tensor(self, tensor: ShardedTensor) -> torch.Tensor: member in TestDistributedReshardOnLoad
H A Dtest_file_system_checkpoint_cpu.py195 def load_tensor(self, tensor: ShardedTensor) -> torch.Tensor: member in TestDistributedReshardOnLoad
/aosp_15_r20/external/pytorch/torch/package/
H A Dpackage_importer.py241 def load_tensor(dtype, size, key, location, restore_location): function
/aosp_15_r20/external/pytorch/torch/
H A Dserialization.py1467 def load_tensor(dtype, numel, key, location): function