MCPcopy Create free account
hub / github.com/apache/tvm / visit_expr

Method visit_expr

python/tvm/relax/testing/ast_printer.py:56–65  ·  view source on GitHub ↗
(self, expr: relax.Expr)

Source from the content-addressed store, hash-verified

54 self.include_call_attrs = include_call_attrs
55
56 def visit_expr(self, expr: relax.Expr) -> str:
57 # extend so we also dispatch to bindings and binding blocks,
58 # a little silly but IRFunctor hasn't been ported to Python
59 if isinstance(expr, relax.DataflowBlock):
60 return self.visit_dataflow_block_(expr)
61 if isinstance(expr, relax.BindingBlock):
62 return self.visit_binding_block_(expr)
63 if isinstance(expr, relax.Binding):
64 return self.visit_binding_(expr)
65 return super().visit_expr(expr)
66
67 def indent(self, text: str) -> str:
68 """

Callers 15

visit_function_Method · 0.95
visit_call_Method · 0.95
visit_seq_expr_Method · 0.95
visit_if_Method · 0.95
visit_tuple_getitem_Method · 0.95
visit_struct_info_Method · 0.95
visit_match_cast_Method · 0.95
visit_var_binding_Method · 0.95
dump_astFunction · 0.95
transform_moduleMethod · 0.45
transform_moduleMethod · 0.45
collectMethod · 0.45

Calls 3

visit_dataflow_block_Method · 0.95
visit_binding_block_Method · 0.95
visit_binding_Method · 0.95

Tested by

no test coverage detected