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