1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
| #pragma once
| #include <pybind11/pybind11.h>
| #include <pybind11/stl.h>
| #include <torch/csrc/jit/script/module.h>
|
| namespace py = pybind11;
|
| namespace torch {
| namespace jit {
| namespace script {
|
| inline c10::optional<Module> as_module(const py::object& obj) {
| if (py::isinstance(
| obj, py::module::import("torch.jit").attr("ScriptModule"))) {
| return py::cast<Module>(obj.attr("_c"));
| }
| return c10::nullopt;
| }
|
| } // namespace script
| } // namespace jit
| } // namespace torch
|
|
