A simple visitor wrapper around ExprFunctor. Recursively visit the content. More...

#include <expr_functor.h>

Inheritance diagram for tvm::relax::ExprVisitor:

Collaboration diagram for tvm::relax::ExprVisitor:

Public Member Functions
void	VisitExpr (const Expr &expr) override
	Generic dispatcher for Expr. More...

void	VisitExpr_ (const ConstantNode *op) override

void	VisitExpr_ (const TupleNode *op) override

void	VisitExpr_ (const VarNode *op) override

void	VisitExpr_ (const DataflowVarNode *op) override

void	VisitExpr_ (const ShapeExprNode *op) override

void	VisitExpr_ (const ExternFuncNode *op) override

void	VisitExpr_ (const GlobalVarNode *op) override

void	VisitExpr_ (const FunctionNode *op) override

void	VisitExpr_ (const CallNode *op) override

void	VisitExpr_ (const SeqExprNode *op) override

void	VisitExpr_ (const IfNode *op) override

void	VisitExpr_ (const OpNode *op) override

void	VisitExpr_ (const TupleGetItemNode *op) override

void	VisitExpr_ (const PrimValueNode *op) override

void	VisitExpr_ (const StringImmNode *op) override

void	VisitExpr_ (const DataTypeImmNode *op) override

virtual void	VisitBinding (const Binding &binding)
	Generic dispatcher for bindings. More...

virtual void	VisitBinding_ (const VarBindingNode *binding)

virtual void	VisitBinding_ (const MatchCastNode *binding)

virtual void	VisitBinding_ (const VarBindingNode binding, const ConstantNode val)

virtual void	VisitBinding_ (const VarBindingNode binding, const TupleNode val)

virtual void	VisitBinding_ (const VarBindingNode binding, const VarNode val)

virtual void	VisitBinding_ (const VarBindingNode binding, const DataflowVarNode val)

virtual void	VisitBinding_ (const VarBindingNode binding, const ShapeExprNode val)

virtual void	VisitBinding_ (const VarBindingNode binding, const ExternFuncNode val)

virtual void	VisitBinding_ (const VarBindingNode binding, const GlobalVarNode val)

virtual void	VisitBinding_ (const VarBindingNode binding, const FunctionNode val)

virtual void	VisitBinding_ (const VarBindingNode binding, const CallNode val)

virtual void	VisitBinding_ (const VarBindingNode binding, const SeqExprNode val)

virtual void	VisitBinding_ (const VarBindingNode binding, const IfNode val)

virtual void	VisitBinding_ (const VarBindingNode binding, const OpNode val)

virtual void	VisitBinding_ (const VarBindingNode binding, const TupleGetItemNode val)

virtual void	VisitBinding_ (const VarBindingNode binding, const PrimValueNode val)

virtual void	VisitBinding_ (const VarBindingNode binding, const StringImmNode val)

virtual void	VisitBinding_ (const VarBindingNode binding, const DataTypeImmNode val)

virtual void	VisitBindingBlock (const BindingBlock &block)
	Generic dispatcher for binding blocks. More...

virtual void	VisitBindingBlock_ (const BindingBlockNode *block)

virtual void	VisitBindingBlock_ (const DataflowBlockNode *block)

virtual void	VisitVarDef (const Var &var)
	Generic dispatcher for visiting the var definition site. More...

virtual void	VisitExprDepStructInfoField (const StructInfo &struct_info)
	Visit struct_info may recursively contain Expr/PrimExpr. More...

virtual void	VisitVarDef_ (const VarNode *var)

virtual void	VisitVarDef_ (const DataflowVarNode *var)

virtual void	VisitSpan (const Span &span)

virtual void	VisitPrimExpr (const PrimExpr &expr)

Detailed Description

A simple visitor wrapper around ExprFunctor. Recursively visit the content.

Member Function Documentation

◆ VisitBinding()

virtual void tvm::relax::ExprVisitor::VisitBinding ( const Binding & binding )

virtual

Generic dispatcher for bindings.

Parameters

binding The binding to be visited.

◆ VisitBinding_() [1/18]

virtual void tvm::relax::ExprVisitor::VisitBinding_ ( const MatchCastNode * binding )

virtual

◆ VisitBinding_() [2/18]

virtual void tvm::relax::ExprVisitor::VisitBinding_ ( const VarBindingNode * binding )

virtual

◆ VisitBinding_() [3/18]

virtual void tvm::relax::ExprVisitor::VisitBinding_	(	const VarBindingNode *	binding,
		const CallNode *	val
	)

virtual

◆ VisitBinding_() [4/18]

virtual void tvm::relax::ExprVisitor::VisitBinding_	(	const VarBindingNode *	binding,
		const ConstantNode *	val
	)

virtual

◆ VisitBinding_() [5/18]

virtual void tvm::relax::ExprVisitor::VisitBinding_	(	const VarBindingNode *	binding,
		const DataflowVarNode *	val
	)

virtual

◆ VisitBinding_() [6/18]

virtual void tvm::relax::ExprVisitor::VisitBinding_	(	const VarBindingNode *	binding,
		const DataTypeImmNode *	val
	)

virtual

◆ VisitBinding_() [7/18]

virtual void tvm::relax::ExprVisitor::VisitBinding_	(	const VarBindingNode *	binding,
		const ExternFuncNode *	val
	)

virtual

◆ VisitBinding_() [8/18]

virtual void tvm::relax::ExprVisitor::VisitBinding_	(	const VarBindingNode *	binding,
		const FunctionNode *	val
	)

virtual

◆ VisitBinding_() [9/18]

virtual void tvm::relax::ExprVisitor::VisitBinding_	(	const VarBindingNode *	binding,
		const GlobalVarNode *	val
	)

virtual

◆ VisitBinding_() [10/18]

virtual void tvm::relax::ExprVisitor::VisitBinding_	(	const VarBindingNode *	binding,
		const IfNode *	val
	)

virtual

◆ VisitBinding_() [11/18]

virtual void tvm::relax::ExprVisitor::VisitBinding_	(	const VarBindingNode *	binding,
		const OpNode *	val
	)

virtual

◆ VisitBinding_() [12/18]

virtual void tvm::relax::ExprVisitor::VisitBinding_	(	const VarBindingNode *	binding,
		const PrimValueNode *	val
	)

virtual

◆ VisitBinding_() [13/18]

virtual void tvm::relax::ExprVisitor::VisitBinding_	(	const VarBindingNode *	binding,
		const SeqExprNode *	val
	)

virtual

◆ VisitBinding_() [14/18]

virtual void tvm::relax::ExprVisitor::VisitBinding_	(	const VarBindingNode *	binding,
		const ShapeExprNode *	val
	)

virtual

◆ VisitBinding_() [15/18]

virtual void tvm::relax::ExprVisitor::VisitBinding_	(	const VarBindingNode *	binding,
		const StringImmNode *	val
	)

virtual

◆ VisitBinding_() [16/18]

virtual void tvm::relax::ExprVisitor::VisitBinding_	(	const VarBindingNode *	binding,
		const TupleGetItemNode *	val
	)

virtual

◆ VisitBinding_() [17/18]

virtual void tvm::relax::ExprVisitor::VisitBinding_	(	const VarBindingNode *	binding,
		const TupleNode *	val
	)

virtual

◆ VisitBinding_() [18/18]

virtual void tvm::relax::ExprVisitor::VisitBinding_	(	const VarBindingNode *	binding,
		const VarNode *	val
	)

virtual

◆ VisitBindingBlock()

virtual void tvm::relax::ExprVisitor::VisitBindingBlock ( const BindingBlock & block )

virtual

Generic dispatcher for binding blocks.

Parameters

block The binding block to be visited.

◆ VisitBindingBlock_() [1/2]

virtual void tvm::relax::ExprVisitor::VisitBindingBlock_ ( const BindingBlockNode * block )

virtual

◆ VisitBindingBlock_() [2/2]

virtual void tvm::relax::ExprVisitor::VisitBindingBlock_ ( const DataflowBlockNode * block )

virtual

◆ VisitExpr()

void tvm::relax::ExprVisitor::VisitExpr ( const Expr & expr )

override

Generic dispatcher for Expr.

Parameters

expr	The expr to be visited.

◆ VisitExpr_() [1/16]

void tvm::relax::ExprVisitor::VisitExpr_ ( const CallNode * op )

override

◆ VisitExpr_() [2/16]

void tvm::relax::ExprVisitor::VisitExpr_ ( const ConstantNode * op )

override

◆ VisitExpr_() [3/16]

void tvm::relax::ExprVisitor::VisitExpr_ ( const DataflowVarNode * op )

override

◆ VisitExpr_() [4/16]

void tvm::relax::ExprVisitor::VisitExpr_ ( const DataTypeImmNode * op )

override

◆ VisitExpr_() [5/16]

void tvm::relax::ExprVisitor::VisitExpr_ ( const ExternFuncNode * op )

override

◆ VisitExpr_() [6/16]

void tvm::relax::ExprVisitor::VisitExpr_ ( const FunctionNode * op )

override

◆ VisitExpr_() [7/16]

void tvm::relax::ExprVisitor::VisitExpr_ ( const GlobalVarNode * op )

override

◆ VisitExpr_() [8/16]

void tvm::relax::ExprVisitor::VisitExpr_ ( const IfNode * op )

override

◆ VisitExpr_() [9/16]

void tvm::relax::ExprVisitor::VisitExpr_ ( const OpNode * op )

override

◆ VisitExpr_() [10/16]

void tvm::relax::ExprVisitor::VisitExpr_ ( const PrimValueNode * op )

override

◆ VisitExpr_() [11/16]

void tvm::relax::ExprVisitor::VisitExpr_ ( const SeqExprNode * op )

override

◆ VisitExpr_() [12/16]

void tvm::relax::ExprVisitor::VisitExpr_ ( const ShapeExprNode * op )

override

◆ VisitExpr_() [13/16]

void tvm::relax::ExprVisitor::VisitExpr_ ( const StringImmNode * op )

override

◆ VisitExpr_() [14/16]

void tvm::relax::ExprVisitor::VisitExpr_ ( const TupleGetItemNode * op )

override

◆ VisitExpr_() [15/16]

void tvm::relax::ExprVisitor::VisitExpr_ ( const TupleNode * op )

override

◆ VisitExpr_() [16/16]

void tvm::relax::ExprVisitor::VisitExpr_ ( const VarNode * op )

override

◆ VisitExprDepStructInfoField()

virtual void tvm::relax::ExprVisitor::VisitExprDepStructInfoField ( const StructInfo & struct_info )

virtual

Visit struct_info may recursively contain Expr/PrimExpr.

By default, this function recurse into struct info such as TensorStructInfo and ShapeStructInfo and call VisitExpr/VisitPrimExpr accordingly. It does not recurse into FunctionStructInfo as it does not contain Expr defined in the current scope.

Pass writers can overload this function to change to other behaviors. For example, if we are not interested in Expr in StructInfo, we can override this function by a no-op.

Parameters

struct_info Input struct info field.

◆ VisitPrimExpr()

virtual void tvm::relax::ExprVisitor::VisitPrimExpr ( const PrimExpr & expr )

virtual

◆ VisitSpan()

virtual void tvm::relax::ExprVisitor::VisitSpan ( const Span & span )

virtual

◆ VisitVarDef()

virtual void tvm::relax::ExprVisitor::VisitVarDef ( const Var & var )

virtual

Generic dispatcher for visiting the var definition site.

Parameters

var	The var to be visited.

Note: VisitExpr_(const VarNode*) will only visit the usage site of an Var

◆ VisitVarDef_() [1/2]

virtual void tvm::relax::ExprVisitor::VisitVarDef_ ( const DataflowVarNode * var )

virtual

◆ VisitVarDef_() [2/2]

virtual void tvm::relax::ExprVisitor::VisitVarDef_ ( const VarNode * var )

virtual

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

include/tvm/relax/expr_functor.h

Public Member Functions

Detailed Description

Member Function Documentation

◆ VisitBinding()

◆ VisitBinding_() [1/18]

◆ VisitBinding_() [2/18]

◆ VisitBinding_() [3/18]

◆ VisitBinding_() [4/18]

◆ VisitBinding_() [5/18]

◆ VisitBinding_() [6/18]

◆ VisitBinding_() [7/18]

◆ VisitBinding_() [8/18]

◆ VisitBinding_() [9/18]

◆ VisitBinding_() [10/18]

◆ VisitBinding_() [11/18]

◆ VisitBinding_() [12/18]

◆ VisitBinding_() [13/18]

◆ VisitBinding_() [14/18]

◆ VisitBinding_() [15/18]

◆ VisitBinding_() [16/18]

◆ VisitBinding_() [17/18]

◆ VisitBinding_() [18/18]

◆ VisitBindingBlock()

◆ VisitBindingBlock_() [1/2]

◆ VisitBindingBlock_() [2/2]

◆ VisitExpr()

◆ VisitExpr_() [1/16]

◆ VisitExpr_() [2/16]

◆ VisitExpr_() [3/16]

◆ VisitExpr_() [4/16]

◆ VisitExpr_() [5/16]

◆ VisitExpr_() [6/16]

◆ VisitExpr_() [7/16]

◆ VisitExpr_() [8/16]

◆ VisitExpr_() [9/16]

◆ VisitExpr_() [10/16]

◆ VisitExpr_() [11/16]

◆ VisitExpr_() [12/16]

◆ VisitExpr_() [13/16]

◆ VisitExpr_() [14/16]

◆ VisitExpr_() [15/16]

◆ VisitExpr_() [16/16]

◆ VisitExprDepStructInfoField()

◆ VisitPrimExpr()

◆ VisitSpan()

◆ VisitVarDef()

◆ VisitVarDef_() [1/2]

◆ VisitVarDef_() [2/2]