tvm
All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros
Public Member Functions | Protected Attributes | List of all members
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 TypePatternNode *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/21]

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

◆ VisitDFPattern_() [2/21]

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

◆ VisitDFPattern_() [3/21]

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

◆ VisitDFPattern_() [4/21]

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

◆ VisitDFPattern_() [5/21]

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

◆ VisitDFPattern_() [6/21]

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

◆ VisitDFPattern_() [7/21]

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

◆ VisitDFPattern_() [8/21]

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

◆ VisitDFPattern_() [9/21]

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

◆ VisitDFPattern_() [10/21]

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

◆ VisitDFPattern_() [11/21]

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

◆ VisitDFPattern_() [12/21]

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

◆ VisitDFPattern_() [13/21]

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

◆ VisitDFPattern_() [14/21]

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

◆ VisitDFPattern_() [15/21]

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

◆ VisitDFPattern_() [16/21]

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

◆ VisitDFPattern_() [17/21]

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

◆ VisitDFPattern_() [18/21]

void tvm::relax::DFPatternVisitor::VisitDFPattern_ ( const TypePatternNode op)
override

◆ VisitDFPattern_() [19/21]

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

◆ VisitDFPattern_() [20/21]

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

◆ VisitDFPattern_() [21/21]

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: