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

Method visit_var_

python/tvm/relax/transform/lazy_transform_params.py:263–279  ·  view source on GitHub ↗
(self, var: relax.Var)

Source from the content-addressed store, hash-verified

261 return output
262
263 def visit_var_(self, var: relax.Var) -> relax.Expr:
264 if var in self.params:
265 index = self.params[var]
266 get_item_result = self.builder_.emit(
267 relax.Call(
268 relax.ExternFunc(self.func_creator.fget_item),
269 self.func_creator.extra_get_item_params + [relax.PrimValue(index)],
270 None,
271 [relax.ObjectStructInfo()],
272 )
273 )
274 match_cast = relax.MatchCast(var, get_item_result, var.struct_info)
275 self.builder_.emit_normalized(match_cast)
276
277 del self.params[var]
278
279 return super().visit_var_(var)
280
281 def visit_tuple_getitem_(self, node: relax.TupleGetItem) -> relax.Expr:
282 sinfo = node.struct_info

Callers

nothing calls this directly

Calls 3

emit_normalizedMethod · 0.80
emitMethod · 0.45
visit_var_Method · 0.45

Tested by

no test coverage detected