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

#include <stmt_functor.h>

Inheritance diagram for tvm::tirx::StmtExprVisitor:

Collaboration diagram for tvm::tirx::StmtExprVisitor:

Protected Member Functions
void	VisitExpr (const PrimExpr &e) override
	Visitor to Exprs, can be overriden to do recursive changes to Exprs. More...

void	VisitExpr_ (const BufferLoadNode *op) override

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

Protected Member Functions inherited from tvm::tirx::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

Protected Member Functions inherited from tvm::tirx::StmtVisitor
virtual void	VisitBufferDef (const Buffer &buffer, bool alloc_data)
	Visit buffer at definition site (AllocBuffer, DeclBuffer, SBlock alloc_buffers). Visits buffer shape, strides, elem_offset via VisitExpr. More...

virtual void	VisitBufferUse (const Buffer &buffer)
	Visit buffer at use site (BufferStore, BufferLoad, SBlock reads/writes). By default, this is a no-op, as buffer fields (shape, strides, elem_offset) are visited at their definition site. More...

void	VisitStmt_ (const BindNode *op) override

void	VisitStmt_ (const AttrStmtNode *op) override

void	VisitStmt_ (const IfThenElseNode *op) override

void	VisitStmt_ (const ForNode *op) override

void	VisitStmt_ (const WhileNode *op) override

void	VisitStmt_ (const AllocBufferNode *op) override

void	VisitStmt_ (const DeclBufferNode *op) override

void	VisitStmt_ (const BufferStoreNode *op) override

void	VisitStmt_ (const AssertStmtNode *op) override

void	VisitStmt_ (const SeqStmtNode *op) override

void	VisitStmt_ (const EvaluateNode *op) override

void	VisitStmt_ (const SBlockNode *op) override

void	VisitStmt_ (const SBlockRealizeNode *op) override

Detailed Description

Visitor that recursively visit stmts and exprs on them.

Member Function Documentation

◆ VisitExpr()

void tvm::tirx::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::tirx::StmtVisitor.