|
|
|
@ -18,11 +18,11 @@ limitations under the License. */
|
|
|
|
|
|
|
|
|
|
#include "paddle/framework/backward.h"
|
|
|
|
|
#include "paddle/framework/op_registry.h"
|
|
|
|
|
#include "paddle/framework/tensor_py.h"
|
|
|
|
|
#include "paddle/operators/net_op.h"
|
|
|
|
|
#include "paddle/operators/recurrent_op.h"
|
|
|
|
|
#include "paddle/platform/enforce.h"
|
|
|
|
|
#include "paddle/platform/place.h"
|
|
|
|
|
#include "paddle/pybind/tensor_py.h"
|
|
|
|
|
#include "paddle/string/to_string.h"
|
|
|
|
|
#include "pybind11/numpy.h"
|
|
|
|
|
#include "pybind11/pybind11.h"
|
|
|
|
@ -135,7 +135,8 @@ All parameter, weight, gradient are variables in Paddle.
|
|
|
|
|
py::return_value_policy::reference)
|
|
|
|
|
.def("find_var", &Scope::FindVar, py::return_value_policy::reference)
|
|
|
|
|
.def(py::init<>())
|
|
|
|
|
.def("new_scope", [](Scope &self) -> Scope * { return &self.NewScope(); },
|
|
|
|
|
.def("new_scope",
|
|
|
|
|
[](Scope &self) -> Scope * { return &self.NewScope(); },
|
|
|
|
|
py::return_value_policy::reference)
|
|
|
|
|
.def("drop_kids", &Scope::DropKids);
|
|
|
|
|
|
|
|
|
@ -223,8 +224,10 @@ All parameter, weight, gradient are variables in Paddle.
|
|
|
|
|
retv->SetType("plain_net");
|
|
|
|
|
return retv;
|
|
|
|
|
})
|
|
|
|
|
.def("append_op", [](operators::NetOp &self,
|
|
|
|
|
const OperatorBase &op) { self.AppendOp(op); })
|
|
|
|
|
.def("append_op",
|
|
|
|
|
[](operators::NetOp &self, const OperatorBase &op) {
|
|
|
|
|
self.AppendOp(op);
|
|
|
|
|
})
|
|
|
|
|
.def("complete_add_op", &operators::NetOp::CompleteAddOp)
|
|
|
|
|
.def("complete_add_op", [](std::shared_ptr<operators::NetOp> &self) {
|
|
|
|
|
self->CompleteAddOp();
|
|
|
|
@ -244,10 +247,9 @@ All parameter, weight, gradient are variables in Paddle.
|
|
|
|
|
auto rnn_op = OpRegistry::CreateOp(desc);
|
|
|
|
|
return static_cast<operators::RecurrentOp *>(rnn_op.release());
|
|
|
|
|
})
|
|
|
|
|
.def("set_stepnet", [](operators::RecurrentOp &self,
|
|
|
|
|
const operators::NetOp &net) -> void {
|
|
|
|
|
self.set_stepnet(net.Clone());
|
|
|
|
|
});
|
|
|
|
|
.def("set_stepnet",
|
|
|
|
|
[](operators::RecurrentOp &self, const operators::NetOp &net)
|
|
|
|
|
-> void { self.set_stepnet(net.Clone()); });
|
|
|
|
|
|
|
|
|
|
m.def("unique_integer", UniqueIntegerGenerator);
|
|
|
|
|
|