mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-21 05:34:18 +08:00
2c7c286fa4
Fixes #ISSUE_NUMBER Pull Request resolved: https://github.com/pytorch/pytorch/pull/129055 Approved by: https://github.com/r-barnes
28 lines
832 B
C++
28 lines
832 B
C++
#pragma once
|
|
|
|
#include <torch/csrc/jit/serialization/unpickler.h>
|
|
#include <memory>
|
|
|
|
namespace caffe2::serialize {
|
|
class PyTorchStreamReader;
|
|
} // namespace caffe2::serialize
|
|
|
|
namespace torch::jit {
|
|
|
|
TORCH_API IValue readArchiveAndTensors(
|
|
const std::string& archive_name,
|
|
const std::string& pickle_prefix,
|
|
const std::string& tensor_prefix,
|
|
std::optional<TypeResolver> type_resolver,
|
|
std::optional<ObjLoader> obj_loader,
|
|
std::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(
|
|
const std::shared_ptr<caffe2::serialize::ReadAdapterInterface>& rai);
|
|
|
|
} // namespace torch::jit
|