提交 1bfa0e13 编写于 作者: F fengjiayi

Add `Prepend` for BlockDescBind

上级 08e99006
...@@ -104,6 +104,12 @@ public: ...@@ -104,6 +104,12 @@ public:
return ops_.back().get(); return ops_.back().get();
} }
OpDescBind *PrependOp() {
need_update_ = true;
ops_.emplace_front(new OpDescBind());
return ops_.front().get();
}
void Sync() { void Sync() {
if (need_update_) { if (need_update_) {
auto &op_field = *this->desc_->mutable_ops(); auto &op_field = *this->desc_->mutable_ops();
...@@ -224,6 +230,9 @@ void BindBlockDesc(py::module &m) { ...@@ -224,6 +230,9 @@ void BindBlockDesc(py::module &m) {
.def("append_op", .def("append_op",
&BlockDescBind::AppendOp, &BlockDescBind::AppendOp,
py::return_value_policy::reference) py::return_value_policy::reference)
.def("prepend_op",
&BlockDescBind::PrependOp,
py::return_value_policy::reference)
.def("new_var", .def("new_var",
&BlockDescBind::NewVar, &BlockDescBind::NewVar,
py::return_value_policy::reference); py::return_value_policy::reference);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册