Home
last modified time | relevance | path

Searched defs:StorageReader (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/torch/distributed/checkpoint/
H A Dstate_dict_loader.py31 storage_reader: StorageReader,
57 storage_reader: Optional[StorageReader] = None,
188 storage_reader: StorageReader,
240 storage_reader: Optional[StorageReader] = None,
H A D_storage_utils.py9 storage: Union[StorageReader, StorageWriter, None],
H A D_checkpointer.py35 storage_reader: StorageReader,
H A Doptimizer.py217 storage_reader: StorageReader,
H A Dstorage.py166 class StorageReader(abc.ABC): class