FM / serialization

/usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc/jit/serialization/

/usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc/jit/serialization

Name	Size	Mode	Actions
callstack_debug_info_serialization.h	2624	0644	edit dl rm
export.h	9839	0644	edit dl rm
import.h	3350	0644	edit dl rm
import_export_constants.h	670	0644	edit dl rm
import_export_functions.h	406	0644	edit dl rm
import_export_helpers.h	709	0644	edit dl rm
import_legacy.h	496	0644	edit dl rm
import_read.h	783	0644	edit dl rm
import_source.h	3342	0644	edit dl rm
onnx.h	229	0644	edit dl rm
pickle.h	3087	0644	edit dl rm
pickler.h	8588	0644	edit dl rm
python_print.h	1180	0644	edit dl rm
source_range_serialization.h	1198	0644	edit dl rm
source_range_serialization_impl.h	685	0644	edit dl rm
storage_context.h	2484	0644	edit dl rm
type_name_uniquer.h	779	0644	edit dl rm
unpickler.h	6434	0644	edit dl rm

Edit: /usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc/jit/serialization/import.h (3350B)

#pragma once

#include <ATen/core/ivalue.h>
#include <caffe2/serialize/inline_container.h>
#include <torch/csrc/jit/api/module.h>
#include <torch/csrc/jit/ir/ir.h>
#include <torch/csrc/jit/serialization/unpickler.h>

#include <istream>

namespace caffe2 {
namespace serialize {
class ReadAdapterInterface;
} // namespace serialize
} // namespace caffe2

namespace torch {
namespace jit {

TORCH_API Module import_ir_module(
    std::shared_ptr<CompilationUnit> cu,
    const std::string& filename,
    c10::optional<c10::Device> device = c10::nullopt);

TORCH_API Module import_ir_module(
    std::shared_ptr<CompilationUnit> cu,
    std::istream& in,
    c10::optional<c10::Device> device = c10::nullopt);

TORCH_API Module import_ir_module(
    std::shared_ptr<CompilationUnit> cu,
    std::unique_ptr<caffe2::serialize::ReadAdapterInterface> rai,
    c10::optional<c10::Device> device = c10::nullopt);

TORCH_API Module import_ir_module(
    std::shared_ptr<CompilationUnit> cu,
    const std::string& filename,
    c10::optional<c10::Device> device,
    ExtraFilesMap& extra_files);

// For reading unified serialization format from torch.Package
TORCH_API Module import_ir_module(
    std::shared_ptr<CompilationUnit> cu,
    std::shared_ptr<caffe2::serialize::PyTorchStreamReader> reader,
    std::shared_ptr<torch::jit::DeserializationStorageContext> storage_context,
    c10::optional<at::Device> device,
    std::string ts_id /* torchscript identifier inside package */);

TORCH_API Module import_ir_module(
    std::shared_ptr<CompilationUnit> cu,
    std::istream& in,
    c10::optional<c10::Device> device,
    ExtraFilesMap& extra_files);

TORCH_API Module import_ir_module(
    std::shared_ptr<CompilationUnit> cu,
    std::unique_ptr<caffe2::serialize::ReadAdapterInterface> rai,
    c10::optional<c10::Device> device,
    ExtraFilesMap& extra_files);

/// Loads a serialized `Module` from the given `istream`.
///
/// The istream must contain a serialized `Module`, exported via
/// `torch::jit::ExportModule` in C++.
TORCH_API Module
load(std::istream& in, c10::optional<c10::Device> device = c10::nullopt);

TORCH_API Module load(
    std::istream& in,
    c10::optional<c10::Device> device,
    ExtraFilesMap& extra_files);

/// Loads a serialized `Module` from the given `filename`.
///
/// The file stored at the location given in `filename` must contain a
/// serialized `Module`, exported either via `ScriptModule.save()` in
/// Python or `torch::jit::ExportModule` in C++.
TORCH_API Module load(
    const std::string& filename,
    c10::optional<c10::Device> device = c10::nullopt);

TORCH_API Module load(
    const std::string& filename,
    c10::optional<c10::Device> device,
    ExtraFilesMap& extra_files);

/// Loads a serialized `Module` from the given shared_ptr `rai`.
///
/// The reader adapter, which is for customized input stream, must contain a
/// serialized `Module`, exported either via `ScriptModule.save()` in
/// Python or `torch::jit::ExportModule` in C++.
TORCH_API Module load(
    std::shared_ptr<caffe2::serialize::ReadAdapterInterface> rai,
    c10::optional<c10::Device> device = c10::nullopt);

TORCH_API Module load(
    std::shared_ptr<caffe2::serialize::ReadAdapterInterface> rai,
    c10::optional<c10::Device> device,
    ExtraFilesMap& extra_files);

} // namespace jit
} // namespace torch

#pragma once #include #include #include #include #include #include namespace caffe2 { namespace serialize { class ReadAdapterInterface; } // namespace serialize } // namespace caffe2 namespace torch { namespace jit { TORCH_API Module import_ir_module( std::shared_ptr cu, const std::string& filename, c10::optional device = c10::nullopt); TORCH_API Module import_ir_module( std::shared_ptr cu, std::istream& in, c10::optional device = c10::nullopt); TORCH_API Module import_ir_module( std::shared_ptr cu, std::unique_ptr rai, c10::optional device = c10::nullopt); TORCH_API Module import_ir_module( std::shared_ptr cu, const std::string& filename, c10::optional device, ExtraFilesMap& extra_files); // For reading unified serialization format from torch.Package TORCH_API Module import_ir_module( std::shared_ptr cu, std::shared_ptr reader, std::shared_ptr storage_context, c10::optional device, std::string ts_id /* torchscript identifier inside package */); TORCH_API Module import_ir_module( std::shared_ptr cu, std::istream& in, c10::optional device, ExtraFilesMap& extra_files); TORCH_API Module import_ir_module( std::shared_ptr cu, std::unique_ptr rai, c10::optional device, ExtraFilesMap& extra_files); /// Loads a serialized `Module` from the given `istream`. /// /// The istream must contain a serialized `Module`, exported via /// `torch::jit::ExportModule` in C++. TORCH_API Module load(std::istream& in, c10::optional device = c10::nullopt); TORCH_API Module load( std::istream& in, c10::optional device, ExtraFilesMap& extra_files); /// Loads a serialized `Module` from the given `filename`. /// /// The file stored at the location given in `filename` must contain a /// serialized `Module`, exported either via `ScriptModule.save()` in /// Python or `torch::jit::ExportModule` in C++. TORCH_API Module load( const std::string& filename, c10::optional device = c10::nullopt); TORCH_API Module load( const std::string& filename, c10::optional device, ExtraFilesMap& extra_files); /// Loads a serialized `Module` from the given shared_ptr `rai`. /// /// The reader adapter, which is for customized input stream, must contain a /// serialized `Module`, exported either via `ScriptModule.save()` in /// Python or `torch::jit::ExportModule` in C++. TORCH_API Module load( std::shared_ptr rai, c10::optional device = c10::nullopt); TORCH_API Module load( std::shared_ptr rai, c10::optional device, ExtraFilesMap& extra_files); } // namespace jit } // namespace torch