/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_legacy.h (496B)
#pragma once
#include
namespace caffe2 {
namespace serialize {
class PyTorchStreamReader;
} // namespace serialize
} // namespace caffe2
namespace torch {
namespace jit {
struct CompilationUnit;
// Deserializes a model in legacy format.
Module LEGACY_deserialize(
std::shared_ptr cu,
std::shared_ptr reader,
const c10::optional& device);
} // namespace jit
} // namespace torch