Home
last modified time | relevance | path

Searched refs:PyTorchFileReader (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/pytorch/torch/package/
H A Dpackage_importer.py96 file_or_buffer: Union[str, torch._C.PyTorchFileReader, os.PathLike, BinaryIO], argument
115 if isinstance(file_or_buffer, torch._C.PyTorchFileReader):
121 self.zip_reader = torch._C.PyTorchFileReader(self.filename)
126 self.zip_reader = torch._C.PyTorchFileReader(file_or_buffer)
250 if isinstance(self.zip_reader, torch._C.PyTorchFileReader):
/aosp_15_r20/external/pytorch/
H A DCODEOWNERS155 # subscribing for PyTorchFileWriter/PyTorchFileReader changes
/aosp_15_r20/external/pytorch/torch/jit/
H A D_script.py384 if not isinstance(importer.zip_reader, torch._C.PyTorchFileReader):
/aosp_15_r20/external/pytorch/torch/
H A Dserialization.py499 super().__init__(torch._C.PyTorchFileReader(name_or_buffer))
/aosp_15_r20/external/pytorch/torch/_C/
H A D__init__.pyi.in717 reader: PyTorchFileReader,
1417 class PyTorchFileReader:
/aosp_15_r20/external/pytorch/test/
H A Dtest_jit.py12777 reader = torch._C.PyTorchFileReader(filename)
12792 torch._C.PyTorchFileReader(filename)
12801 torch._C.PyTorchFileReader(f)