diff --git a/python/tvm/topi/utils.py b/python/tvm/topi/utils.py index e24c499a0e98..c9ba1d23dbaf 100644 --- a/python/tvm/topi/utils.py +++ b/python/tvm/topi/utils.py @@ -185,6 +185,8 @@ def get_const_tuple(in_tuple): out_tuple : tuple of int The output. """ + if not isinstance(in_tuple, (tuple, list)): + return (in_tuple,) ret = [] ana = None for elem in in_tuple: