Remove shared_ptr holder for VarBase

test=develop
revert-15207-remove_op_handle_lock_and_fix_var
minqiyang 6 years ago
parent c8d1a8e909
commit cded24768c

@ -125,8 +125,7 @@ PYBIND11_MODULE(core, m) {
m.add_object("_cleanup", m.add_object("_cleanup",
py::capsule([]() { ScopePool::Instance().Clear(); })); py::capsule([]() { ScopePool::Instance().Clear(); }));
py::class_<imperative::VarBase, std::shared_ptr<imperative::VarBase>>( py::class_<imperative::VarBase>(m, "VarBase", R"DOC()DOC")
m, "VarBase", R"DOC()DOC")
// .def(py::init<>()) // .def(py::init<>())
.def(py::init<bool>(), py::arg("stop_gradient") = false) .def(py::init<bool>(), py::arg("stop_gradient") = false)
.def("_run_backward", .def("_run_backward",

Loading…
Cancel
Save