expr_functor.h File Reference

Functors for tir expressions. More...

#include <tvm/node/functor.h>
#include <tvm/tir/expr.h>
#include <utility>

Include dependency graph for expr_functor.h:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes
class	tvm::tir::ExprFunctor< FType >
	A dynamical functor that dispatches on in the first Expr argument. You can use this as a more powerful Visitor, since it allows you to define function signatures of Visit Function. More...
class	tvm::tir::ExprFunctor< R(const PrimExpr &n, Args...)>
class	tvm::tir::ExprVisitor
	ExprVisitor. More...
class	tvm::tir::ExprMutator
	ExprMutator that mutates expressions. More...

Namespaces
	tvm
	runtime implementation for LibTorch/TorchScript.
	tvm::tir

Macros
#define	EXPR_FUNCTOR_DEFAULT   { return VisitExprDefault_(op, std::forward<Args>(args)...); }
#define	IR_EXPR_FUNCTOR_DISPATCH(OP)

Detailed Description

Functors for tir expressions.

Macro Definition Documentation

EXPR_FUNCTOR_DEFAULT

#define EXPR_FUNCTOR_DEFAULT   { return VisitExprDefault_(op, std::forward<Args>(args)...); }

IR_EXPR_FUNCTOR_DISPATCH

#define IR_EXPR_FUNCTOR_DISPATCH

(

OP

)

Value:

vtable.template set_dispatch<OP>([](const ObjectRef& n, TSelf* self, Args... args) {     \
    return self->VisitExpr_(static_cast<const OP*>(n.get()), std::forward<Args>(args)...); \
  });