def METHOD_NAME(self):
    for gv, func in mod.functions.items():
        if isinstance(func, relax.Function):
            updated_func = self.visit_expr(func)
            self.builder_.update_func(gv, updated_func)
    new_mod = self.builder_.get()
    new_mod = new_mod.with_attrs(mod.attrs) if mod.attrs else new_mod
    return new_mod