diff --git a/python/tvm/relay/expr_functor.py b/python/tvm/relay/expr_functor.py index 48941b2b23b9..05e0feb0c354 100644 --- a/python/tvm/relay/expr_functor.py +++ b/python/tvm/relay/expr_functor.py @@ -251,7 +251,7 @@ def visit_tuple_getitem(self, op): new_tuple_value = self.visit(op.tuple_value) if new_tuple_value == op.tuple_value: return op - return TupleGetItem(new_tuple_value, op.index) + return TupleGetItem(new_tuple_value, op.index, span=op.span) def visit_global_var(self, gvar): return gvar