(self, op: relax.Function)
| 141 | return self.build_expr(op, "GlobalVar", name_hint=wrap_quotes(op.name_hint)) |
| 142 | |
| 143 | def visit_function_(self, op: relax.Function) -> str: |
| 144 | fields = { |
| 145 | "params": self.build_list(map(self.visit_expr, op.params)), |
| 146 | "body": self.visit_expr(op.body), |
| 147 | "ret_struct_info": self.visit_struct_info_(op.ret_struct_info), |
| 148 | "is_pure": op.is_pure, |
| 149 | } |
| 150 | if op.attrs: |
| 151 | fields["attrs"] = self.build_list( |
| 152 | map( |
| 153 | lambda kv: f"{wrap_quotes(str(kv[0]))}: {wrap_quotes(str(kv[1]))}", |
| 154 | op.attrs.items(), |
| 155 | ), |
| 156 | open_tok="{", |
| 157 | close_tok="}", |
| 158 | ) |
| 159 | return self.build_expr(op, "Function", **fields) |
| 160 | |
| 161 | def visit_call_(self, op: relax.Call) -> str: |
| 162 | fields = { |
nothing calls this directly
no test coverage detected