1 | #pragma once |
---|---|
2 | |
3 | #include <torch/csrc/jit/serialization/unpickler.h> |
4 | #include <memory> |
5 | |
6 | namespace caffe2 { |
7 | namespace serialize { |
8 | class PyTorchStreamReader; |
9 | } // namespace serialize |
10 | } // namespace caffe2 |
11 | |
12 | namespace torch { |
13 | namespace jit { |
14 | |
15 | TORCH_API IValue readArchiveAndTensors( |
16 | const std::string& archive_name, |
17 | const std::string& pickle_prefix, |
18 | const std::string& tensor_prefix, |
19 | c10::optional<TypeResolver> type_resolver, |
20 | c10::optional<ObjLoader> obj_loader, |
21 | c10::optional<at::Device> device, |
22 | caffe2::serialize::PyTorchStreamReader& stream_reader, |
23 | c10::TypePtr (*type_parser)(const std::string&) = |
24 | Unpickler::defaultTypeParser, |
25 | std::shared_ptr<DeserializationStorageContext> storage_context = nullptr); |
26 | |
27 | bool check_zip_file( |
28 | std::shared_ptr<caffe2::serialize::ReadAdapterInterface> rai); |
29 | |
30 | } // namespace jit |
31 | } // namespace torch |
32 |