tvm
•All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros
Protected Member Functions | List of all members
tvm::tir::StmtExprVisitor Class Reference

Visitor that recursively visit stmts and exprs on them. More...

#include <stmt_functor.h>

Inheritance diagram for tvm::tir::StmtExprVisitor:
Collaboration diagram for tvm::tir::StmtExprVisitor:

Protected Member Functions

void VisitExpr (const PrimExpr &e) override
 Visitor to Exprs, can be overriden to do recursive changes to Exprs. More...
 
- Protected Member Functions inherited from tvm::tir::StmtVisitor
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 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
 
- Protected Member Functions inherited from tvm::tir::ExprVisitor
void VisitExpr_ (const VarNode *op) override
 
void VisitExpr_ (const SizeVarNode *op) override
 
void VisitExpr_ (const BufferLoadNode *op) override
 
void VisitExpr_ (const ProducerLoadNode *op) override
 
void VisitExpr_ (const LetNode *op) override
 
void VisitExpr_ (const CallNode *op) override
 
void VisitExpr_ (const AddNode *op) override
 
void VisitExpr_ (const SubNode *op) override
 
void VisitExpr_ (const MulNode *op) override
 
void VisitExpr_ (const DivNode *op) override
 
void VisitExpr_ (const ModNode *op) override
 
void VisitExpr_ (const FloorDivNode *op) override
 
void VisitExpr_ (const FloorModNode *op) override
 
void VisitExpr_ (const MinNode *op) override
 
void VisitExpr_ (const MaxNode *op) override
 
void VisitExpr_ (const EQNode *op) override
 
void VisitExpr_ (const NENode *op) override
 
void VisitExpr_ (const LTNode *op) override
 
void VisitExpr_ (const LENode *op) override
 
void VisitExpr_ (const GTNode *op) override
 
void VisitExpr_ (const GENode *op) override
 
void VisitExpr_ (const AndNode *op) override
 
void VisitExpr_ (const OrNode *op) override
 
void VisitExpr_ (const ReduceNode *op) override
 
void VisitExpr_ (const CastNode *op) override
 
void VisitExpr_ (const NotNode *op) override
 
void VisitExpr_ (const SelectNode *op) override
 
void VisitExpr_ (const RampNode *op) override
 
void VisitExpr_ (const BroadcastNode *op) override
 
void VisitExpr_ (const ShuffleNode *op) override
 
void VisitExpr_ (const IntImmNode *op) override
 
void VisitExpr_ (const FloatImmNode *op) override
 
void VisitExpr_ (const StringImmNode *op) override
 
void VisitExpr_ (const AnyNode *op) override
 

Detailed Description

Visitor that recursively visit stmts and exprs on them.

Member Function Documentation

◆ VisitExpr()

void tvm::tir::StmtExprVisitor::VisitExpr ( const PrimExpr &  e)
inlineoverrideprotectedvirtual

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 from tvm::tir::StmtVisitor.

The documentation for this class was generated from the following file:
Generated by doxygen 1.9.1