Function _eval_op

python/tvm/script/parser/core/evaluator.py:531–560

Operation expression evaluation implementation for TVMScript parser. Parameters ---------- op : doc.AST The root node of AST tree node of operation expression to evaluate. values : List[Any] The list of values of operands. Returns ------- res : Any

(
    op: doc.AST,
    values: list[Any],
)

def _eval_op(
    op: doc.AST,
    values: list[Any],
):
    """Operation expression evaluation implementation for TVMScript parser.

    Parameters
    ----------
    op : doc.AST
        The root node of AST tree node of operation expression to evaluate.

    values : List[Any]
        The list of values of operands.

    Returns
    -------
    res : Any
        The evaluation result.
    """
    op_type = type(op)  # pylint: disable=protected-access
    for i, v in enumerate(values):
        v_type = getattr(type(v), "_dispatch_type", None)
        if v_type is None:
            continue
        f = dispatch.get_op(
            operand_type=v_type, op_node_type=op_type, operand_index=i, default=None
        )
        if f is not None:
            return f(*values)
    return DEFAULT_OP[op_type](*values)


def _eval_assign(