Function get_const_tuple

python/tvm/topi/utils.py:176–208

Verifies input tuple is IntImm or Var, returns tuple of int or Var. Parameters ---------- in_tuple : tuple of Expr The input. Returns ------- out_tuple : tuple of int The output.

(in_tuple)

def get_const_tuple(in_tuple):
    """Verifies input tuple is IntImm or Var, returns tuple of int or Var.

    Parameters
    ----------
    in_tuple : tuple of Expr
        The input.

    Returns
    -------
    out_tuple : tuple of int
        The output.
    """
    if isinstance(in_tuple, te.tensor.Tensor):
        raise TypeError(
            "get_const_tuple expects a tuple-like shape (e.g., tensor.shape), "
            "but got a te.Tensor. Did you mean get_const_tuple(tensor.shape)?"
        )
    ret = []
    ana = None
    for elem in in_tuple:
        if isinstance(elem, tvm.tirx.Var):
            ret.append(elem)
        elif not isinstance(elem, tvm.tirx.IntImm | int):
            ana = tvm.arith.Analyzer() if ana is None else ana
            elem = ana.simplify(elem)
            if not isinstance(elem, tvm.tirx.IntImm):
                ret.append(elem)
            else:
                ret.append(get_const_int(elem))
        else:
            ret.append(get_const_int(elem))
    return tuple(ret)


def const_vector(vector, name="const_vector"):