Method visit_var_def

python/tvm/relax/expr_functor.py:258–264

(self, var: Var)

            raise TypeError(f"Invalid type: {type(block)}")

    def visit_var_def(self, var: Var):
        if isinstance(var, DataflowVar):
            self.visit_dataflow_var_def_(var)
        elif isinstance(var, Var):
            self.visit_var_def_(var)
        else:
            raise TypeError(f"Invalid type: {type(var)}")


@tvm_ffi.register_object("expr_functor.PyExprVisitor")