/usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc/jit/python
Edit: /usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc/jit/python/module_python.h (685B)
#pragma once
#include
#include
#include
namespace py = pybind11;
namespace torch {
namespace jit {
inline c10::optional as_module(const py::object& obj) {
if (py::isinstance(
obj, py::module::import("torch.jit").attr("ScriptModule"))) {
return py::cast(obj.attr("_c"));
}
return c10::nullopt;
}
inline c10::optional