tvm::relax::DFPatternVisitor Class Reference

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

#include <dataflow_pattern_functor.h>

Inheritance diagram for tvm::relax::DFPatternVisitor:

Collaboration diagram for tvm::relax::DFPatternVisitor:

Public Member Functions
void	VisitDFPattern (const DFPattern &pattern) override
void	VisitDFPattern_ (const OrPatternNode *op) override
void	VisitDFPattern_ (const AndPatternNode *op) override
void	VisitDFPattern_ (const NotPatternNode *op) override
void	VisitDFPattern_ (const AttrPatternNode *op) override
void	VisitDFPattern_ (const CallPatternNode *op) override
void	VisitDFPattern_ (const ConstantPatternNode *op) override
void	VisitDFPattern_ (const DataTypePatternNode *op) override
void	VisitDFPattern_ (const ExprPatternNode *op) override
void	VisitDFPattern_ (const FunctionPatternNode *op) override
void	VisitDFPattern_ (const ShapePatternNode *op) override
void	VisitDFPattern_ (const TupleGetItemPatternNode *op) override
void	VisitDFPattern_ (const TuplePatternNode *op) override
void	VisitDFPattern_ (const StructInfoPatternNode *op) override
void	VisitDFPattern_ (const WildcardPatternNode *op) override
void	VisitDFPattern_ (const VarPatternNode *op) override
void	VisitDFPattern_ (const DataflowVarPatternNode *op) override
void	VisitDFPattern_ (const GlobalVarPatternNode *op) override
void	VisitDFPattern_ (const ExternFuncPatternNode *op) override
void	VisitDFPattern_ (const PrimArrPatternNode *op) override
void	VisitDFPattern_ (const UnorderedTuplePatternNode *op) override

Protected Attributes
std::unordered_set< const Object * >	visited_

Detailed Description

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

DFPatternVisitor treats the Pattern as dataflow graph,and only visit each Expr node once.

Member Function Documentation

VisitDFPattern()

void tvm::relax::DFPatternVisitor::VisitDFPattern ( const DFPattern &  pattern )

override

VisitDFPattern_() [1/20]

void tvm::relax::DFPatternVisitor::VisitDFPattern_ ( const AndPatternNode *  op )

override

VisitDFPattern_() [2/20]

void tvm::relax::DFPatternVisitor::VisitDFPattern_ ( const AttrPatternNode *  op )

override

VisitDFPattern_() [3/20]

void tvm::relax::DFPatternVisitor::VisitDFPattern_ ( const CallPatternNode *  op )

override

VisitDFPattern_() [4/20]

void tvm::relax::DFPatternVisitor::VisitDFPattern_ ( const ConstantPatternNode *  op )

override

VisitDFPattern_() [5/20]

void tvm::relax::DFPatternVisitor::VisitDFPattern_ ( const DataflowVarPatternNode *  op )

override

VisitDFPattern_() [6/20]

void tvm::relax::DFPatternVisitor::VisitDFPattern_ ( const DataTypePatternNode *  op )

override

VisitDFPattern_() [7/20]

void tvm::relax::DFPatternVisitor::VisitDFPattern_ ( const ExprPatternNode *  op )

override

VisitDFPattern_() [8/20]

void tvm::relax::DFPatternVisitor::VisitDFPattern_ ( const ExternFuncPatternNode *  op )

override

VisitDFPattern_() [9/20]

void tvm::relax::DFPatternVisitor::VisitDFPattern_ ( const FunctionPatternNode *  op )

override

VisitDFPattern_() [10/20]

void tvm::relax::DFPatternVisitor::VisitDFPattern_ ( const GlobalVarPatternNode *  op )

override

VisitDFPattern_() [11/20]

void tvm::relax::DFPatternVisitor::VisitDFPattern_ ( const NotPatternNode *  op )

override

VisitDFPattern_() [12/20]

void tvm::relax::DFPatternVisitor::VisitDFPattern_ ( const OrPatternNode *  op )

override

VisitDFPattern_() [13/20]

void tvm::relax::DFPatternVisitor::VisitDFPattern_ ( const PrimArrPatternNode *  op )

override

VisitDFPattern_() [14/20]

void tvm::relax::DFPatternVisitor::VisitDFPattern_ ( const ShapePatternNode *  op )

override

VisitDFPattern_() [15/20]

void tvm::relax::DFPatternVisitor::VisitDFPattern_ ( const StructInfoPatternNode *  op )

override

VisitDFPattern_() [16/20]

void tvm::relax::DFPatternVisitor::VisitDFPattern_ ( const TupleGetItemPatternNode *  op )

override

VisitDFPattern_() [17/20]

void tvm::relax::DFPatternVisitor::VisitDFPattern_ ( const TuplePatternNode *  op )

override

VisitDFPattern_() [18/20]

void tvm::relax::DFPatternVisitor::VisitDFPattern_ ( const UnorderedTuplePatternNode *  op )

override

VisitDFPattern_() [19/20]

void tvm::relax::DFPatternVisitor::VisitDFPattern_ ( const VarPatternNode *  op )

override

VisitDFPattern_() [20/20]

void tvm::relax::DFPatternVisitor::VisitDFPattern_ ( const WildcardPatternNode *  op )

override

Member Data Documentation

visited_

std::unordered_set<const Object*> tvm::relax::DFPatternVisitor::visited_

protected

---

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

• include/tvm/relax/dataflow_pattern_functor.h