tvm::relax::StructInfoMutator Class Reference

StructInfoMutator that mutates struct info. More...

#include <struct_info_functor.h>

Inheritance diagram for tvm::relax::StructInfoMutator:

Collaboration diagram for tvm::relax::StructInfoMutator:

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

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

Detailed Description

StructInfoMutator that mutates struct info.

Member Function Documentation

VisitStructInfo_() [1/7]

StructInfo tvm::relax::StructInfoMutator::VisitStructInfo_ ( const distributed::DTensorStructInfoNode *  op )

override

VisitStructInfo_() [2/7]

StructInfo tvm::relax::StructInfoMutator::VisitStructInfo_ ( const FuncStructInfoNode *  op )

override

VisitStructInfo_() [3/7]

StructInfo tvm::relax::StructInfoMutator::VisitStructInfo_ ( const ObjectStructInfoNode *  op )

override

VisitStructInfo_() [4/7]

StructInfo tvm::relax::StructInfoMutator::VisitStructInfo_ ( const PrimStructInfoNode *  op )

override

VisitStructInfo_() [5/7]

StructInfo tvm::relax::StructInfoMutator::VisitStructInfo_ ( const ShapeStructInfoNode *  op )

override

VisitStructInfo_() [6/7]

StructInfo tvm::relax::StructInfoMutator::VisitStructInfo_ ( const TensorStructInfoNode *  op )

override

VisitStructInfo_() [7/7]

StructInfo tvm::relax::StructInfoMutator::VisitStructInfo_ ( const TupleStructInfoNode *  op )

override

VisitStructInfoExprField() [1/2]

virtual Expr tvm::relax::StructInfoMutator::VisitStructInfoExprField ( const Expr &  expr )

inlineprotectedvirtual

VisitStructInfoExprField() [2/2]

virtual PrimExpr tvm::relax::StructInfoMutator::VisitStructInfoExprField ( const PrimExpr &  expr )

inlineprotectedvirtual

---

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

• include/tvm/relax/struct_info_functor.h