(self, op: tvm.ir.Op)
| 217 | return self.build_expr(op, "DataTypeImm", value=op.value) |
| 218 | |
| 219 | def visit_op_(self, op: tvm.ir.Op) -> str: |
| 220 | # TODO: List other attributes? |
| 221 | # op is not actually a Relax expr and does not have |
| 222 | # struct_info fields, so we don't use build_expr here |
| 223 | return self.build_ast_node("Op", name=wrap_quotes(op.name)) |
| 224 | |
| 225 | def visit_prim_expr_(self, prim_expr: PrimExpr) -> str: |
| 226 | # TODO: We may want to print PrimExpr ASTs, but this is a simplification for now |
nothing calls this directly
no test coverage detected