1#pragma once
2
3#include <torch/csrc/jit/serialization/unpickler.h>
4#include <memory>
5
6namespace caffe2 {
7namespace serialize {
8class PyTorchStreamReader;
9} // namespace serialize
10} // namespace caffe2
11
12namespace torch {
13namespace jit {
14
15TORCH_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
27bool check_zip_file(
28 std::shared_ptr<caffe2::serialize::ReadAdapterInterface> rai);
29
30} // namespace jit
31} // namespace torch
32