提交 afeb01f7 编写于 作者: Y Yu Yang

Stash

上级 b941865d
...@@ -114,6 +114,14 @@ public: ...@@ -114,6 +114,14 @@ public:
std::string DebugString() { return this->Proto()->DebugString(); } std::string DebugString() { return this->Proto()->DebugString(); }
struct SetAttrDescVisitor : public boost::static_visitor<void> {
explicit SetAttrDescVisitor(OpDesc::Attr *attr) : attr_(attr) {}
OpDesc::Attr *attr_;
void operator()(int v) { attr_->set_i(v); }
void operator()(float v) { attr_->set_f(v); }
void operator()(const std::string &v) { attr_->set_s(v); }
};
void Sync() { void Sync() {
if (need_update_) { if (need_update_) {
this->op_desc_.mutable_inputs()->Clear(); this->op_desc_.mutable_inputs()->Clear();
...@@ -130,6 +138,13 @@ public: ...@@ -130,6 +138,13 @@ public:
VectorToRepeated(opt.second, output->mutable_arguments()); VectorToRepeated(opt.second, output->mutable_arguments());
} }
this->op_desc_.mutable_attrs()->Clear();
for (auto &attr : attrs_) {
auto *attr_desc = op_desc_.add_attrs();
attr_desc->set_name(attr.first);
attr_desc->set_type(static_cast<AttrType>(attr.second.which() - 1));
}
need_update_ = false; need_update_ = false;
} }
} }
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册