tqchen commented on code in PR #15916:
URL: https://github.com/apache/tvm/pull/15916#discussion_r1372345395
##########
include/tvm/relax/expr_functor.h:
##########
@@ -278,6 +278,37 @@ class ExprVisitor : public ExprFunctor<void(const Expr&)> {
virtual void VisitSpan(const Span& span);
virtual void VisitPrimExpr(const PrimExpr& expr);
+ /*!
+ * \brief Look up the value bound to a variable.
+ * \param var The var to be looked up.
+ * \return The value bound to the input \p var.
+ * \note For function parameters, this function returns NullOpt.
+ */
+ inline Optional<Expr> LookupBinding(const Var& var) {
Review Comment:
I think we should instead enforce call_tir to be a more restricted form
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
