Skip to content

T
tic
Commit 26ceaac8 by 雾雨魔理沙 Committed by Tianqi Chen
Options

save (#2015)

parent 74314d27
Showing with 3 additions and 0 deletions
include/tvm/relay/expr.h
...@@ -134,6 +134,7 @@ class VarNode : public ExprNode { ...@@ -134,6 +134,7 @@ class VarNode : public ExprNode {
void VisitAttrs(tvm::AttrVisitor* v) final { void VisitAttrs(tvm::AttrVisitor* v) final {
v->Visit("name_hint", &name_hint); v->Visit("name_hint", &name_hint);
v->Visit("type_annotation", &type_annotation); v->Visit("type_annotation", &type_annotation);
v->Visit("span", &span);
v->Visit("_checked_type_", &checked_type_); v->Visit("_checked_type_", &checked_type_);
} }
...@@ -161,6 +162,7 @@ class GlobalVarNode : public ExprNode { ...@@ -161,6 +162,7 @@ class GlobalVarNode : public ExprNode {
void VisitAttrs(tvm::AttrVisitor* v) final { void VisitAttrs(tvm::AttrVisitor* v) final {
v->Visit("name_hint", &name_hint); v->Visit("name_hint", &name_hint);
v->Visit("span", &span);
v->Visit("_checked_type_", &checked_type_); v->Visit("_checked_type_", &checked_type_);
} }
...@@ -377,6 +379,7 @@ class TupleGetItemNode : public ExprNode { ...@@ -377,6 +379,7 @@ class TupleGetItemNode : public ExprNode {
void VisitAttrs(tvm::AttrVisitor* v) final { void VisitAttrs(tvm::AttrVisitor* v) final {
v->Visit("tuple_value", &tuple); v->Visit("tuple_value", &tuple);
v->Visit("index", &index); v->Visit("index", &index);
v->Visit("span", &span);
v->Visit("_checked_type_", &checked_type_); v->Visit("_checked_type_", &checked_type_);
} }
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment