Home
last modified time | relevance | path

Searched refs:PyTorchStreamReader (Results 1 – 22 of 22) sorted by relevance

/aosp_15_r20/external/pytorch/caffe2/serialize/
H A Dinline_container.cc64 auto self = static_cast<PyTorchStreamReader*>(pOpaque); in istream_read_func()
111 size_t PyTorchStreamReader::read(uint64_t pos, char* buf, size_t n) { in read()
116 PyTorchStreamReader::PyTorchStreamReader(const std::string& file_name) in PyTorchStreamReader() function in caffe2::serialize::PyTorchStreamReader
123 PyTorchStreamReader::PyTorchStreamReader(std::istream* in) in PyTorchStreamReader() function in caffe2::serialize::PyTorchStreamReader
130 PyTorchStreamReader::PyTorchStreamReader( in PyTorchStreamReader() function in caffe2::serialize::PyTorchStreamReader
136 void PyTorchStreamReader::init() { in init()
234 void PyTorchStreamReader::valid(const char* what, const char* info) { in valid()
283 bool PyTorchStreamReader::hasRecord(const std::string& name) { in hasRecord()
304 std::vector<std::string> PyTorchStreamReader::getAllRecords() { in getAllRecords()
336 size_t PyTorchStreamReader::getRecordID(const std::string& name) { in getRecordID()
[all …]
H A Dinline_container.h120 friend class PyTorchStreamReader; variable
123 class TORCH_API PyTorchStreamReader final {
125 explicit PyTorchStreamReader(const std::string& file_name);
126 explicit PyTorchStreamReader(std::istream* in);
127 explicit PyTorchStreamReader(std::shared_ptr<ReadAdapterInterface> in);
173 ~PyTorchStreamReader();
H A Dinline_container_test.cc60 PyTorchStreamReader reader(&iss); in TEST()
149 PyTorchStreamReader reader(&iss); in TEST()
232 PyTorchStreamReader reader(&iss); in TEST()
293 PyTorchStreamReader reader(&iss); in TEST()
339 PyTorchStreamReader reader(&iss); in TEST()
382 PyTorchStreamReader reader(&iss); in TEST()
407 PyTorchStreamReader reader(&iss); in TEST()
465 PyTorchStreamReader reader(fileName); in TEST_P()
/aosp_15_r20/external/pytorch/torch/csrc/jit/mobile/compatibility/
H A Dmodel_compatibility.cpp27 using caffe2::serialize::PyTorchStreamReader;
32 PyTorchStreamReader& stream_reader) { in readArchive()
63 std::vector<IValue> get_bytecode_ivalues(PyTorchStreamReader& reader) { in get_bytecode_ivalues()
100 PyTorchStreamReader reader(std::move(rai)); in _get_model_bytecode_version_zip()
141 PyTorchStreamReader& reader); // Forward Declare
160 PyTorchStreamReader reader(std::move(rai)); in _get_model_operator_version()
164 uint64_t _get_model_operator_version(PyTorchStreamReader& reader) { in _get_model_operator_version()
192 PyTorchStreamReader reader(std::move(rai)); in _get_model_ops_and_info()
275 PyTorchStreamReader reader(std::move(rai)); in _get_mobile_model_contained_types()
333 PyTorchStreamReader reader(std::move(rai)); in get()
H A Dbackport_manager.cpp17 using caffe2::serialize::PyTorchStreamReader;
37 PyTorchStreamReader& reader, in selective_copy()
173 PyTorchStreamReader reader_bytecode(&input_model); in update_bytecode_version()
291 PyTorchStreamReader reader(&input_model_stream); in backport_v5_to_v4()
390 auto reader = std::make_shared<PyTorchStreamReader>(rai); in backport_v6_to_v5()
455 auto reader = std::make_shared<PyTorchStreamReader>(rai); in backport_v7_to_v6()
528 auto reader = std::make_shared<PyTorchStreamReader>(rai); in backport_v8_to_v7()
686 PyTorchStreamReader last_model_reader(&output_model_stream); in backport()
H A Dmodel_compatibility.h13 class PyTorchStreamReader; variable
42 caffe2::serialize::PyTorchStreamReader& reader);
46 caffe2::serialize::PyTorchStreamReader& stream_reader);
/aosp_15_r20/external/pytorch/torch/csrc/jit/serialization/
H A Dimport.cpp39 using caffe2::serialize::PyTorchStreamReader;
116 std::shared_ptr<PyTorchStreamReader> reader) in ScriptModuleDeserializer()
133 std::shared_ptr<PyTorchStreamReader> reader, in ScriptModuleDeserializer()
161 std::shared_ptr<PyTorchStreamReader> reader_;
372 auto reader = std::make_unique<PyTorchStreamReader>(&in); in import_ir_module()
385 std::shared_ptr<PyTorchStreamReader> reader, in import_ir_module()
419 auto reader = std::make_unique<PyTorchStreamReader>(filename); in import_ir_module()
456 auto reader = std::make_shared<PyTorchStreamReader>(std::move(rai)); in import_ir_module()
533 auto reader = std::make_unique<PyTorchStreamReader>(std::move(rai)); in _load_jit_module_from_bytes()
H A Dimport_export_helpers.h7 class PyTorchStreamReader; variable
24 caffe2::serialize::PyTorchStreamReader& reader,
H A Dimport_read.h7 class PyTorchStreamReader; variable
19 caffe2::serialize::PyTorchStreamReader& stream_reader,
H A Dpickle.cpp127 caffe2::serialize::PyTorchStreamReader reader( in pickle_load()
148 caffe2::serialize::PyTorchStreamReader reader( in pickle_load_obj()
H A Dimport_export_helpers.cpp24 caffe2::serialize::PyTorchStreamReader& reader, in findSourceInArchiveFromQualifier()
H A Dimport_read.cpp15 caffe2::serialize::PyTorchStreamReader& stream_reader, in readArchiveAndTensors()
H A Dimport.h47 std::shared_ptr<caffe2::serialize::PyTorchStreamReader> reader,
/aosp_15_r20/external/pytorch/torch/csrc/jit/mobile/
H A Dimport_data.cpp24 using caffe2::serialize::PyTorchStreamReader;
34 explicit IValueUnpickler(std::unique_ptr<PyTorchStreamReader> reader);
44 std::unique_ptr<PyTorchStreamReader> reader_;
47 IValueUnpickler::IValueUnpickler(std::unique_ptr<PyTorchStreamReader> reader) in IValueUnpickler()
162 auto reader = std::make_unique<PyTorchStreamReader>(std::move(rai)); in load_parameters_from_zip()
H A Dimport.cpp86 using caffe2::serialize::PyTorchStreamReader;
162 caffe2::serialize::PyTorchStreamReader& stream_reader) { in isTensorInBytecodeArchive()
191 std::unique_ptr<PyTorchStreamReader> reader,
218 std::unique_ptr<PyTorchStreamReader> reader_{};
230 std::unique_ptr<PyTorchStreamReader> reader, in BytecodeDeserializer()
521 auto reader = std::make_unique<PyTorchStreamReader>(std::move(rai)); in _load_for_mobile_impl()
693 auto reader = std::make_unique<PyTorchStreamReader>(std::move(rai)); in _load_extra_only_for_mobile()
H A Ddebug_info.h24 std::unique_ptr<caffe2::serialize::PyTorchStreamReader>& reader,
H A Dimport.h89 caffe2::serialize::PyTorchStreamReader& stream_reader);
H A Ddebug_info.cpp114 std::unique_ptr<caffe2::serialize::PyTorchStreamReader>& reader, in MobileDebugTable()
/aosp_15_r20/external/pytorch/torch/csrc/jit/python/
H A Dinit.cpp130 using caffe2::serialize::PyTorchStreamReader;
1563 py::class_<PyTorchStreamReader, std::shared_ptr<PyTorchStreamReader>>( in initJITBindings()
1568 return std::make_shared<PyTorchStreamReader>(std::move(adapter)); in initJITBindings()
1572 [](PyTorchStreamReader& self, const std::string& key) { in initJITBindings()
1578 [](PyTorchStreamReader& self, const std::string& key) { in initJITBindings()
1583 [](PyTorchStreamReader& self, in initJITBindings()
1604 .def("serialization_id", &PyTorchStreamReader::serializationId) in initJITBindings()
1607 [](PyTorchStreamReader& self) { return self.getAllRecords(); }) in initJITBindings()
1610 [](PyTorchStreamReader& self, const std::string& key) { in initJITBindings()
H A Dscript_init.cpp1865 std::shared_ptr<caffe2::serialize::PyTorchStreamReader> reader, in initJitScriptBindings()
/aosp_15_r20/external/pytorch/test/cpp/jit/
H A Dtest_save_load.cpp327 caffe2::serialize::PyTorchStreamReader reader(&ss); in TEST()
H A Dtest_lite_interpreter.cpp1006 caffe2::serialize::PyTorchStreamReader(&iss).getAllRecords(); in TEST()