stmt_functor.h File Reference

Functors for tir stmts utility functions to call common functors. More...

#include <tvm/node/functor.h>
#include <tvm/tir/expr.h>
#include <tvm/tir/expr_functor.h>
#include <tvm/tir/function.h>
#include <tvm/tir/stmt.h>
#include <unordered_map>
#include <utility>

Include dependency graph for stmt_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::StmtFunctor< FType >
	Same as ExprFunctor except it is applied on statements. More...
class	tvm::tir::StmtFunctor< R(const Stmt &n, Args... args)>
class	tvm::tir::StmtVisitor
	StmtVisitor. More...
class	tvm::tir::StmtMutator
	StmtMutator that mutates the statements. More...
class	tvm::tir::StmtExprVisitor
	Visitor that recursively visit stmts and exprs on them. More...
class	tvm::tir::StmtExprMutator
	Mutator that recursively mutates stmts and exprs on them. More...

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

Macros
#define	STMT_FUNCTOR_DEFAULT   { return VisitStmtDefault_(op, std::forward<Args>(args)...); }
#define	IR_STMT_FUNCTOR_DISPATCH(OP)

Functions
Stmt	tvm::tir::IRTransform (Stmt stmt, const runtime::PackedFunc &preorder, const runtime::PackedFunc &postorder, Optional< Array< String >> only_enable=NullOpt)
	recursively visit the ir nodes in post DFS order, and transform it More...
void	tvm::tir::PostOrderVisit (const ObjectRef &node, std::function< void(const ObjectRef &)> fvisit)
	Recursively visit the ir in post DFS order node, apply fvisit Each node is guaranteed to be visited only once. More...
Stmt	tvm::tir::Substitute (Stmt stmt, std::function< Optional< PrimExpr >(const Var &var)> vmap)
	Substitute the var specified by vmap. More...
PrimExpr	tvm::tir::Substitute (PrimExpr expr, std::function< Optional< PrimExpr >(const Var &var)> vmap)
	Substitute the var specified by vmap. More...
Array< Range >	tvm::tir::Substitute (const Array< Range > &region, const Map< Var, PrimExpr > &vmap)
	Substitute the var specified by vmap. More...
template<typename T >
auto	tvm::tir::Substitute (T input, const Map< Var, PrimExpr > &value_map)
	Sugar for substitute via a given map. More...
template<typename T >
T	tvm::tir::Substitute (T input, const std::unordered_map< const VarNode *, PrimExpr > &value_map)
	Sugar for substitute via a given map. More...
void	tvm::tir::PreOrderVisit (const ObjectRef &stmt_or_expr, const std::function< bool(const ObjectRef &)> &fvisit)
	Recursively visit the IR in pre DFS order node, apply fvisit. If fvisit returns false, it won't visit the children of the node. More...
PrimFunc	tvm::tir::RenewDefs (const PrimFunc &func)
	Renew the definition nodes for a TIR, including Var, Buffer and IterVar. This pass works as a simple DeepCopy to duplicate a function with different Vars and Buffers but the same behavior. More...

Detailed Description

Functors for tir stmts utility functions to call common functors.

Macro Definition Documentation

IR_STMT_FUNCTOR_DISPATCH

#define IR_STMT_FUNCTOR_DISPATCH

(

OP

)

Value:

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

STMT_FUNCTOR_DEFAULT

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