frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add a pybind for Module::get_functions. (#20594)

Browse Source 
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/20594
ghimport-source-id: 96f8046fd3aed49d8b29312ce2f8d7e0ea5e5787

Differential Revision: D15375132

Pulled By: ZolotukhinM

fbshipit-source-id: 7ad87e29d965e459c49be1be8a08f86ed2a0b4db
This commit is contained in:
Mikhail Zolotukhin
2019-05-17 19:53:56 -07:00
committed by Facebook Github Bot
parent 2af5911a95
commit 986c9eb537
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
torch/csrc/jit/script/init.cpp
View File

@ -312,6 +312,11 @@ void initJitScriptBindings(PyObject* module) {
}, },
py::return_value_policy::reference_internal) py::return_value_policy::reference_internal)
.def("_register_parameter", &Module::register_parameter) .def("_register_parameter", &Module::register_parameter)
.def(
"_get_functions",
[](Module& self) {
return self.class_compilation_unit().get_functions();
})
.def( .def(
"_register_attribute", "_register_attribute",
[](Module& self, std::string name, TypePtr type, py::object value) { [](Module& self, std::string name, TypePtr type, py::object value) {