| /aosp_15_r20/external/pytorch/torch/csrc/jit/mobile/compatibility/ |
| H A D | backport.cpp | 13 using caffe2::serialize::PyTorchStreamWriter;
21 PyTorchStreamWriter& writer,
32 PyTorchStreamWriter writer(writer_func); in _backport_for_mobile()
40 PyTorchStreamWriter writer(output_filename); in _backport_for_mobile()
59 PyTorchStreamWriter writer(writer_func); in _backport_for_mobile()
73 PyTorchStreamWriter writer(output_filename); in _backport_for_mobile()
79 PyTorchStreamWriter& writer, in _backport_for_mobile_impl()
|
| H A D | backport_manager.cpp | 18 using caffe2::serialize::PyTorchStreamWriter;
38 PyTorchStreamWriter& writer, in selective_copy()
82 PyTorchStreamWriter& writer, in write_archive_current()
193 PyTorchStreamWriter writer_bytecode(writer_func); in update_bytecode_version()
313 PyTorchStreamWriter writer(writer_func); in backport_v5_to_v4()
326 auto writeArchiveV4 = [](PyTorchStreamWriter& writer, in backport_v5_to_v4()
611 PyTorchStreamWriter& final_writer, in backport()
|
| H A D | backport_manager.h | 12 class PyTorchStreamWriter; variable
32 caffe2::serialize::PyTorchStreamWriter& final_writer,
|
| /aosp_15_r20/external/pytorch/caffe2/serialize/ |
| H A D | inline_container.cc | 332 PyTorchStreamWriter::getAllWrittenRecords() { in getAllWrittenRecords()
601 auto self = static_cast<PyTorchStreamWriter*>(pOpaque); in ostream_write_func()
624 PyTorchStreamWriter::PyTorchStreamWriter(const std::string& file_name) in PyTorchStreamWriter() function in caffe2::serialize::PyTorchStreamWriter
629 PyTorchStreamWriter::PyTorchStreamWriter( in PyTorchStreamWriter() function in caffe2::serialize::PyTorchStreamWriter
636 void PyTorchStreamWriter::setup(const string& file_name) { in setup()
675 void PyTorchStreamWriter::setMinVersion(const uint64_t version) { in setMinVersion()
679 void PyTorchStreamWriter::writeRecord( in writeRecord()
717 void PyTorchStreamWriter::writeEndOfFile() { in writeEndOfFile()
775 void PyTorchStreamWriter::valid(const char* what, const char* info) { in valid()
790 void PyTorchStreamWriter::writeSerializationId() { in writeSerializationId()
[all …]
|
| H A D | inline_container_test.cc | 21 PyTorchStreamWriter writer([&](const void* b, size_t n) -> size_t { in TEST()
113 PyTorchStreamWriter writer([&](const void* b, size_t n) -> size_t { in TEST()
192 PyTorchStreamWriter writer([&](const void* b, size_t n) -> size_t { in TEST()
255 PyTorchStreamWriter writer([&](const void* b, size_t n) -> size_t { in TEST()
319 PyTorchStreamWriter writer([&](const void* b, size_t n) -> size_t { in TEST()
345 PyTorchStreamWriter writer2([&](const void* b, size_t n) -> size_t { in TEST()
358 PyTorchStreamWriter writer([&](const void* b, size_t n) -> size_t { in TEST()
399 PyTorchStreamWriter writer([&](const void* b, size_t n) -> size_t { in TEST()
443 PyTorchStreamWriter writer([&](const void* b, size_t n) -> size_t { in TEST_P()
|
| H A D | inline_container.h | 206 class TORCH_API PyTorchStreamWriter final {
208 explicit PyTorchStreamWriter(const std::string& archive_name);
209 explicit PyTorchStreamWriter(
235 ~PyTorchStreamWriter();
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/python/ |
| H A D | init.cpp | 131 using caffe2::serialize::PyTorchStreamWriter;
1391 py::class_<PyTorchStreamWriter>(m, "PyTorchFileWriter") in initJITBindings()
1411 return std::make_unique<PyTorchStreamWriter>(std::move(writer_func)); in initJITBindings()
1427 [](PyTorchStreamWriter& self, const std::string& name, size_t size) { in initJITBindings()
1432 [](PyTorchStreamWriter& self, in initJITBindings()
1442 [](PyTorchStreamWriter& self, in initJITBindings()
1456 [](PyTorchStreamWriter& self, in initJITBindings()
1467 [](PyTorchStreamWriter& self, in initJITBindings()
1477 .def("write_end_of_file", &PyTorchStreamWriter::writeEndOfFile) in initJITBindings()
1478 .def("set_min_version", &PyTorchStreamWriter::setMinVersion) in initJITBindings()
[all …]
|
| H A D | script_init.cpp | 1021 .def(py::init<caffe2::serialize::PyTorchStreamWriter&>()) in initJitScriptBindings()
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/serialization/ |
| H A D | export.h | 73 caffe2::serialize::PyTorchStreamWriter& export_writer) in ScriptModuleSerializer()
101 caffe2::serialize::PyTorchStreamWriter& writer_;
187 caffe2::serialize::PyTorchStreamWriter& out);
|
| H A D | pickle.cpp | 82 caffe2::serialize::PyTorchStreamWriter writer( in pickle_save()
|
| H A D | export_module.cpp | 860 caffe2::serialize::PyTorchStreamWriter writer(filename); in ExportModule()
930 caffe2::serialize::PyTorchStreamWriter writer(writer_func); in ExportModule()
|
| H A D | export.cpp | 77 caffe2::serialize::PyTorchStreamWriter& out) { in writeArchiveAndTensors()
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/mobile/train/ |
| H A D | export_data.cpp | 70 caffe2::serialize::PyTorchStreamWriter writer_;
|