|
|
|
@ -104,6 +104,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();
|
|
|
|
@ -224,6 +230,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);
|
|
|
|
|