mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 00:21:07 +01:00
31524bda1f
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/19721 ghimport-source-id: b4f5024adc845a82dc5197d19aab1496bf85089f Reviewed By: jamesr66a Differential Revision: D15078534 Pulled By: zdevito fbshipit-source-id: 408d3a871302c5ac5d6426dc5de567f2188ebf4c
37 lines
1.2 KiB
C++
37 lines
1.2 KiB
C++
#pragma once
|
|
|
|
#include <torch/csrc/jit/ir.h>
|
|
#include <torch/csrc/jit/script/compiler.h>
|
|
#include <torch/csrc/jit/script/module.h>
|
|
|
|
namespace torch {
|
|
namespace jit {
|
|
namespace script {
|
|
// Helpers to define modules and classes from source strings. Used in model
|
|
// saving/loading, so it expects the format produced by the model exporter.
|
|
|
|
// Add the methods defined in `src` to the module `mod`.
|
|
TORCH_API void import_methods(
|
|
const std::shared_ptr<script::Module>& mod,
|
|
const std::string& src,
|
|
const std::vector<at::Tensor>& constant_table);
|
|
|
|
// Defined the list of classes in `src`.
|
|
TORCH_API void import_libs(
|
|
const std::string& src,
|
|
const std::vector<at::Tensor>& constant_table);
|
|
|
|
// Add the functions defined in `src` to the module `mod`.
|
|
// self is passed through the CompilationUnit's define function.
|
|
// If present, it determines the SugaredValue for the first argument
|
|
// and that argument is no longer expected to have type annotations.
|
|
TORCH_API void import_functions(
|
|
CompilationUnit& cu,
|
|
const std::string& src,
|
|
const std::vector<at::Tensor>& constant_table,
|
|
const Self& self=nullptr);
|
|
|
|
} // namespace script
|
|
} // namespace jit
|
|
} // namespace torch
|