未验证 提交 4a4567fc 编写于 作者: X Xin Pan 提交者: GitHub

Merge pull request #12876 from panyx0718/ir5

refine Node
...@@ -142,8 +142,6 @@ class Graph { ...@@ -142,8 +142,6 @@ class Graph {
nodes_.erase(node); nodes_.erase(node);
} }
const ProgramDesc &program() const { return program_; }
private: private:
// This method takes ownership of `node`. // This method takes ownership of `node`.
ir::Node *AddNode(ir::Node *node) { ir::Node *AddNode(ir::Node *node) {
...@@ -154,7 +152,7 @@ class Graph { ...@@ -154,7 +152,7 @@ class Graph {
} }
// NOTE: program_ shouldn't be exposed to user. // NOTE: program_ shouldn't be exposed to user.
const ProgramDesc &program_; const ProgramDesc program_;
std::map<std::string, boost::any> attrs_; std::map<std::string, boost::any> attrs_;
std::map<std::string, std::function<void(void)>> attr_dels_; std::map<std::string, std::function<void(void)>> attr_dels_;
std::map<ir::Node *, std::unique_ptr<ir::Node>> nodes_; std::map<ir::Node *, std::unique_ptr<ir::Node>> nodes_;
......
...@@ -41,8 +41,7 @@ class Node { ...@@ -41,8 +41,7 @@ class Node {
explicit Node(OpDesc* op_desc) explicit Node(OpDesc* op_desc)
: name_(op_desc->Type()), : name_(op_desc->Type()),
var_desc_(nullptr), var_desc_(nullptr),
op_desc_(new OpDesc(*op_desc)), // TODO(panyx0718) the pointer in the op_desc_(new OpDesc(*op_desc, op_desc->Block())),
// original OpDesc might go out.
type_(Type::kOperation) {} type_(Type::kOperation) {}
Type NodeType() const { return type_; } Type NodeType() const { return type_; }
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册