tvm::tir::StmtVisitor Class Reference

StmtVisitor. More...

#include <stmt_functor.h>

Inheritance diagram for tvm::tir::StmtVisitor:

Collaboration diagram for tvm::tir::StmtVisitor:

Protected Member Functions
virtual void	VisitExpr (const PrimExpr &e)
	Visitor to Exprs, can be overriden to do recursive changes to Exprs. More...
void	VisitStmt_ (const AttrStmtNode *op) override
void	VisitStmt_ (const IfThenElseNode *op) override
void	VisitStmt_ (const LetStmtNode *op) override
void	VisitStmt_ (const ForNode *op) override
void	VisitStmt_ (const WhileNode *op) override
void	VisitStmt_ (const AllocateNode *op) override
void	VisitStmt_ (const AllocateConstNode *op) override
void	VisitStmt_ (const DeclBufferNode *op) override
void	VisitStmt_ (const StoreNode *op) override
void	VisitStmt_ (const BufferStoreNode *op) override
void	VisitStmt_ (const BufferRealizeNode *op) override
void	VisitStmt_ (const AssertStmtNode *op) override
void	VisitStmt_ (const ProducerStoreNode *op) override
void	VisitStmt_ (const ProducerRealizeNode *op) override
void	VisitStmt_ (const PrefetchNode *op) override
void	VisitStmt_ (const SeqStmtNode *op) override
void	VisitStmt_ (const EvaluateNode *op) override
void	VisitStmt_ (const BlockNode *op) override
void	VisitStmt_ (const BlockRealizeNode *op) override

Detailed Description

StmtVisitor.

Member Function Documentation

VisitExpr()

virtual void tvm::tir::StmtVisitor::VisitExpr ( const PrimExpr &  e )

inlineprotectedvirtual

Visitor to Exprs, can be overriden to do recursive changes to Exprs.

Note

A common pattern is to call ExprVisitor here, or have a class sub-class both StmtVisitor and ExprVisitor and redirect Visit to ExprMutator::VisitExpr(Expr)

Reimplemented in tvm::tir::StmtExprVisitor.

VisitStmt_() [1/19]

void tvm::tir::StmtVisitor::VisitStmt_ ( const AttrStmtNode *  op )

overrideprotected

VisitStmt_() [2/19]

void tvm::tir::StmtVisitor::VisitStmt_ ( const IfThenElseNode *  op )

overrideprotected

VisitStmt_() [3/19]

void tvm::tir::StmtVisitor::VisitStmt_ ( const LetStmtNode *  op )

overrideprotected

VisitStmt_() [4/19]

void tvm::tir::StmtVisitor::VisitStmt_ ( const ForNode *  op )

overrideprotected

VisitStmt_() [5/19]

void tvm::tir::StmtVisitor::VisitStmt_ ( const WhileNode *  op )

overrideprotected

VisitStmt_() [6/19]

void tvm::tir::StmtVisitor::VisitStmt_ ( const AllocateNode *  op )

overrideprotected

VisitStmt_() [7/19]

void tvm::tir::StmtVisitor::VisitStmt_ ( const AllocateConstNode *  op )

overrideprotected

VisitStmt_() [8/19]

void tvm::tir::StmtVisitor::VisitStmt_ ( const DeclBufferNode *  op )

overrideprotected

VisitStmt_() [9/19]

void tvm::tir::StmtVisitor::VisitStmt_ ( const StoreNode *  op )

overrideprotected

VisitStmt_() [10/19]

void tvm::tir::StmtVisitor::VisitStmt_ ( const BufferStoreNode *  op )

overrideprotected

VisitStmt_() [11/19]

void tvm::tir::StmtVisitor::VisitStmt_ ( const BufferRealizeNode *  op )

overrideprotected

VisitStmt_() [12/19]

void tvm::tir::StmtVisitor::VisitStmt_ ( const AssertStmtNode *  op )

overrideprotected

VisitStmt_() [13/19]

void tvm::tir::StmtVisitor::VisitStmt_ ( const ProducerStoreNode *  op )

overrideprotected

VisitStmt_() [14/19]

void tvm::tir::StmtVisitor::VisitStmt_ ( const ProducerRealizeNode *  op )

overrideprotected

VisitStmt_() [15/19]

void tvm::tir::StmtVisitor::VisitStmt_ ( const PrefetchNode *  op )

overrideprotected

VisitStmt_() [16/19]

void tvm::tir::StmtVisitor::VisitStmt_ ( const SeqStmtNode *  op )

overrideprotected

VisitStmt_() [17/19]

void tvm::tir::StmtVisitor::VisitStmt_ ( const EvaluateNode *  op )

overrideprotected

VisitStmt_() [18/19]

void tvm::tir::StmtVisitor::VisitStmt_ ( const BlockNode *  op )

overrideprotected

VisitStmt_() [19/19]

void tvm::tir::StmtVisitor::VisitStmt_ ( const BlockRealizeNode *  op )

overrideprotected

---

The documentation for this class was generated from the following file:

• include/tvm/tir/stmt_functor.h