/usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc/jit/python
NameSizeModeActions
init.h1510644editdlrm
module_python.h6850644editdlrm
pybind.h81020644editdlrm
pybind_utils.h397110644editdlrm
python_arg_flatten.h35770644editdlrm
python_custom_class.h4710644editdlrm
python_dict.h33450644editdlrm
python_ir.h17180644editdlrm
python_ivalue.h32490644editdlrm
python_list.h54380644editdlrm
python_sugared_value.h105450644editdlrm
python_tracer.h9110644editdlrm
python_tree_views.h1750644editdlrm
script_init.h1770644editdlrm
update_graph_executor_opt.h2240644editdlrm
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 as_object(const py::object& obj) { if (py::isinstance( obj, py::module::import("torch.jit").attr("RecursiveScriptClass"))) { return py::cast(obj.attr("_c")); } return c10::nullopt; } } // namespace jit } // namespace torch