tvm
Protected Member Functions | List of all members
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 BufferStoreNode *op) override
 
void VisitStmt_ (const BufferRealizeNode *op) override
 
void VisitStmt_ (const AssertStmtNode *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/15]

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

◆ VisitStmt_() [2/15]

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

◆ VisitStmt_() [3/15]

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

◆ VisitStmt_() [4/15]

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

◆ VisitStmt_() [5/15]

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

◆ VisitStmt_() [6/15]

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

◆ VisitStmt_() [7/15]

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

◆ VisitStmt_() [8/15]

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

◆ VisitStmt_() [9/15]

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

◆ VisitStmt_() [10/15]

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

◆ VisitStmt_() [11/15]

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

◆ VisitStmt_() [12/15]

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

◆ VisitStmt_() [13/15]

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

◆ VisitStmt_() [14/15]

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

◆ VisitStmt_() [15/15]

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

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