Function dump_ast

python/tvm/relax/testing/ast_printer.py:361–377

Dump an AST in a text format. Can vary the indentation string and choose whether to include type and shape annotations or call attributes.

(
    exp: relax.Expr,
    indent_str="    ",
    include_struct_info_annotations=True,
    include_call_attrs=True,
)

def dump_ast(
    exp: relax.Expr,
    indent_str="    ",
    include_struct_info_annotations=True,
    include_call_attrs=True,
) -> str:
    """
    Dump an AST in a text format.
    Can vary the indentation string and choose whether to include
    type and shape annotations or call attributes.
    """
    printer = ASTPrinter(
        indent_str=indent_str,
        include_struct_info_annotations=include_struct_info_annotations,
        include_call_attrs=include_call_attrs,
    )
    return printer.visit_expr(exp)