save (#2015)

26ceaac8 · 雾雨魔理沙 · Tianqi Chen · 74314d27 · 26ceaac8
Commit 26ceaac8 authored Oct 27, 2018 by 雾雨魔理沙 Committed by Tianqi Chen Oct 27, 2018
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 0 deletions

include/tvm/relay/expr.h
+3 -0

No files found.
--- a/include/tvm/relay/expr.h
+++ b/include/tvm/relay/expr.h
@@ -134,6 +134,7 @@ class VarNode : public ExprNode {
  void VisitAttrs(tvm::AttrVisitor* v) final {
    v->Visit("name_hint", &name_hint);
    v->Visit("type_annotation", &type_annotation);
+    v->Visit("span", &span);
    v->Visit("_checked_type_", &checked_type_);
  }

@@ -161,6 +162,7 @@ class GlobalVarNode : public ExprNode {

  void VisitAttrs(tvm::AttrVisitor* v) final {
    v->Visit("name_hint", &name_hint);
+    v->Visit("span", &span);
    v->Visit("_checked_type_", &checked_type_);
  }

@@ -377,6 +379,7 @@ class TupleGetItemNode : public ExprNode {
  void VisitAttrs(tvm::AttrVisitor* v) final {
    v->Visit("tuple_value", &tuple);
    v->Visit("index", &index);
+    v->Visit("span", &span);
    v->Visit("_checked_type_", &checked_type_);
  }