Merge branch 'feature/pybind_for_protobuf_desc' of github.com:reyoung/Paddle into feature/pybind_for_protobuf_desc

update-doc-pybind
Yu Yang 8 years ago
commit 6cf396b6f3

@ -214,6 +214,12 @@ public:
return ops_.back().get();
}
OpDescBind *PrependOp() {
need_update_ = true;
ops_.emplace_front(new OpDescBind());
return ops_.front().get();
}
void Sync() {
if (need_update_) {
auto &op_field = *this->desc_->mutable_ops();
@ -329,6 +335,9 @@ void BindBlockDesc(py::module &m) {
.def("append_op",
&BlockDescBind::AppendOp,
py::return_value_policy::reference)
.def("prepend_op",
&BlockDescBind::PrependOp,
py::return_value_policy::reference)
.def("new_var",
&BlockDescBind::NewVar,
py::return_value_policy::reference);

Loading…
Cancel
Save