1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
|
#pragma once
#include <torch/csrc/jit/api/module.h>
#include <torch/csrc/jit/ir/ir.h>
#include <functional>
#include <memory>
#include <string>
#include <vector>
namespace torch {
namespace jit {
struct SourceImporterImpl;
// Given a directory of serialized TorchScript sources,
// This class allows the loading of individual named types in source.
// Resolves the dependencies between source files and parses
// the source files as necessary.
using SourceLoader = std::function<std::shared_ptr<Source>(const std::string&)>;
struct TORCH_API SourceImporter {
SourceImporter(
// The compilation unit that will own the imported source
std::shared_ptr<CompilationUnit> cu,
const std::vector<at::IValue>* constant_table,
SourceLoader loader,
size_t version);
TypePtr loadType(const QualifiedName& name) const;
// Add the methods defined in `src` to the module `mod`, using SourceImporter
// to resolve any classes via loadType
void LEGACY_import_methods(
const Module& mod,
const std::shared_ptr<Source>& src);
~SourceImporter();
private:
std::shared_ptr<SourceImporterImpl> pImpl;
};
} // namespace jit
} // namespace torch
|
