mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 12:54:11 +08:00
9fff8155c3
This PR applies clang-tidy readability checks to jit sources and all headers in the code base. `readability-redundant-inline-specifier` is suppressed because it incurs too many changes. `readability-redundant-inline-specifier` is used to detect redundant inline specifiers on function and variable declarations. There are many in-class method definitions that are marked inline. Pull Request resolved: https://github.com/pytorch/pytorch/pull/164652 Approved by: https://github.com/Skylion007
148 lines
4.7 KiB
C++
148 lines
4.7 KiB
C++
#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 <istream>
|
|
|
|
namespace caffe2::serialize {
|
|
class ReadAdapterInterface;
|
|
} // namespace caffe2::serialize
|
|
|
|
namespace torch::jit {
|
|
|
|
class DeserializationStorageContext;
|
|
|
|
TORCH_API Module import_ir_module(
|
|
std::shared_ptr<CompilationUnit> cu,
|
|
const std::string& filename,
|
|
std::optional<c10::Device> device = std::nullopt,
|
|
bool load_debug_files = true);
|
|
|
|
TORCH_API Module import_ir_module(
|
|
std::shared_ptr<CompilationUnit> cu,
|
|
std::istream& in,
|
|
std::optional<c10::Device> device = std::nullopt,
|
|
bool load_debug_files = true);
|
|
|
|
TORCH_API Module import_ir_module(
|
|
std::shared_ptr<CompilationUnit> cu,
|
|
std::unique_ptr<caffe2::serialize::ReadAdapterInterface> rai,
|
|
std::optional<c10::Device> device = std::nullopt,
|
|
bool load_debug_files = true);
|
|
|
|
TORCH_API Module import_ir_module(
|
|
std::shared_ptr<CompilationUnit> cu,
|
|
const std::string& filename,
|
|
std::optional<c10::Device> device,
|
|
ExtraFilesMap& extra_files,
|
|
bool load_debug_files = true,
|
|
bool restore_shapes = false);
|
|
|
|
// 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,
|
|
std::optional<at::Device> device,
|
|
const std::string& ts_id /* torchscript identifier inside package */);
|
|
|
|
TORCH_API Module import_ir_module(
|
|
std::shared_ptr<CompilationUnit> cu,
|
|
std::istream& in,
|
|
std::optional<c10::Device> device,
|
|
ExtraFilesMap& extra_files,
|
|
bool load_debug_files = true,
|
|
bool restore_shapes = false);
|
|
|
|
TORCH_API Module import_ir_module(
|
|
std::shared_ptr<CompilationUnit> cu,
|
|
std::unique_ptr<caffe2::serialize::ReadAdapterInterface> rai,
|
|
std::optional<c10::Device> device,
|
|
ExtraFilesMap& extra_files,
|
|
bool load_debug_files = true);
|
|
|
|
TORCH_API Module import_ir_module(
|
|
std::shared_ptr<CompilationUnit> cu,
|
|
std::shared_ptr<caffe2::serialize::ReadAdapterInterface> rai,
|
|
std::optional<c10::Device> device,
|
|
ExtraFilesMap& extra_files,
|
|
bool load_debug_files = true);
|
|
|
|
/// 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,
|
|
std::optional<c10::Device> device = std::nullopt,
|
|
bool load_debug_files = true);
|
|
|
|
TORCH_API Module load(
|
|
std::istream& in,
|
|
std::optional<c10::Device> device,
|
|
ExtraFilesMap& extra_files,
|
|
bool load_debug_files = true);
|
|
|
|
/// 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,
|
|
std::optional<c10::Device> device = std::nullopt,
|
|
bool load_debug_files = true);
|
|
|
|
TORCH_API Module load(
|
|
const std::string& filename,
|
|
std::optional<c10::Device> device,
|
|
ExtraFilesMap& extra_files,
|
|
bool load_debug_files = true);
|
|
|
|
/// 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,
|
|
std::optional<c10::Device> device = std::nullopt,
|
|
bool load_debug_files = true);
|
|
|
|
TORCH_API Module load(
|
|
std::shared_ptr<caffe2::serialize::ReadAdapterInterface> rai,
|
|
std::optional<c10::Device> device,
|
|
ExtraFilesMap& extra_files,
|
|
bool load_debug_files = true);
|
|
|
|
TORCH_API Module jitModuleFromSourceAndConstants(
|
|
const IValue& ivalue,
|
|
const ExtraFilesMap& source,
|
|
const std::vector<IValue>& constants,
|
|
int32_t version);
|
|
|
|
TORCH_API Module parse_and_initialize_jit_module(
|
|
const std::shared_ptr<char>& data,
|
|
size_t size,
|
|
ExtraFilesMap& extra_files,
|
|
std::optional<at::Device> device = std::nullopt);
|
|
|
|
TORCH_API Module load_jit_module_from_file(
|
|
const std::string& filename,
|
|
ExtraFilesMap& extra_files,
|
|
std::optional<at::Device> device = std::nullopt);
|
|
|
|
TORCH_API Module load_jit_module_from_stream(
|
|
std::istream& in,
|
|
ExtraFilesMap& extra_files,
|
|
std::optional<at::Device> device = std::nullopt);
|
|
|
|
TORCH_API c10::intrusive_ptr<c10::ivalue::Object> ObjLoaderFunc(
|
|
const at::StrongTypePtr& type,
|
|
IValue input);
|
|
|
|
} // namespace torch::jit
|