tvm::relax::StructInfoVisitor Class Reference

A struct info visitor. More...

#include <struct_info_functor.h>

Inheritance diagram for tvm::relax::StructInfoVisitor:

Collaboration diagram for tvm::relax::StructInfoVisitor:

Public Member Functions
void	VisitStructInfo_ (const ObjectStructInfoNode *op) override
void	VisitStructInfo_ (const PrimStructInfoNode *op) override
void	VisitStructInfo_ (const ShapeStructInfoNode *op) override
void	VisitStructInfo_ (const TensorStructInfoNode *op) override
void	VisitStructInfo_ (const distributed::DTensorStructInfoNode *op) override
void	VisitStructInfo_ (const TupleStructInfoNode *op) override
void	VisitStructInfo_ (const FuncStructInfoNode *op) override

Protected Member Functions
virtual void	VisitStructInfoExprField (const Expr &expr)
virtual void	VisitStructInfoExprField (const PrimExpr &expr)

Detailed Description

A struct info visitor.

Member Function Documentation

VisitStructInfo_() [1/7]

void tvm::relax::StructInfoVisitor::VisitStructInfo_ ( const distributed::DTensorStructInfoNode *  op )

override

VisitStructInfo_() [2/7]

void tvm::relax::StructInfoVisitor::VisitStructInfo_ ( const FuncStructInfoNode *  op )

override

VisitStructInfo_() [3/7]

void tvm::relax::StructInfoVisitor::VisitStructInfo_ ( const ObjectStructInfoNode *  op )

override

VisitStructInfo_() [4/7]

void tvm::relax::StructInfoVisitor::VisitStructInfo_ ( const PrimStructInfoNode *  op )

override

VisitStructInfo_() [5/7]

void tvm::relax::StructInfoVisitor::VisitStructInfo_ ( const ShapeStructInfoNode *  op )

override

VisitStructInfo_() [6/7]

void tvm::relax::StructInfoVisitor::VisitStructInfo_ ( const TensorStructInfoNode *  op )

override

VisitStructInfo_() [7/7]

void tvm::relax::StructInfoVisitor::VisitStructInfo_ ( const TupleStructInfoNode *  op )

override

VisitStructInfoExprField() [1/2]

virtual void tvm::relax::StructInfoVisitor::VisitStructInfoExprField ( const Expr &  expr )

inlineprotectedvirtual

VisitStructInfoExprField() [2/2]

virtual void tvm::relax::StructInfoVisitor::VisitStructInfoExprField ( const PrimExpr &  expr )

inlineprotectedvirtual

---

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

• include/tvm/relax/struct_info_functor.h