copy all attributes from broadcast_to, ones, zeros, full

apache · Jul 13, 2020 · 5c5025d · 5c5025d
1 parent ccb44d7
commit 5c5025d
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/python/tvm/relay/_parser.py b/python/tvm/relay/_parser.py
@@ -116,8 +116,6 @@ def __call__(self, args, attrs, type_args):
             attrs = {}
         if self.operator in (op.strided_slice,):
             x = self.operator(*args)
-        elif self.operator in (op.zeros, op.ones, op.full, op.broadcast_to):
-            x = self.operator(*args, dtype=attrs["dtype"])
         else:
             x = self.operator(*args, **{k: self.convert(v) for k, v in attrs.items()})
         if isinstance(x, expr.TupleWrapper):