/usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc/jit/serialization
Edit: /usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc/jit/serialization/import_read.h (783B)
#pragma once
#include
#include
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 type_resolver,
c10::optional obj_loader,
c10::optional device,
caffe2::serialize::PyTorchStreamReader& stream_reader,
std::shared_ptr storage_context = nullptr);
bool check_zip_file(
std::shared_ptr rai);
} // namespace jit
} // namespace torch