tvm
Public Member Functions | List of all members
tvm::relax::ShapeStructInfo Class Reference

Managed reference to ShapeStructInfoNode. More...

#include <struct_info.h>

Inheritance diagram for tvm::relax::ShapeStructInfo:
Collaboration diagram for tvm::relax::ShapeStructInfo:

Public Member Functions

 ShapeStructInfo (Array< PrimExpr > values, Span span=Span())
 Construction with known symbolic shape patterns. More...
 
 ShapeStructInfo (int ndim, Span span=Span())
 Construction with known unknown symbolic shape patterns. More...
 
 TVM_DEFINE_NOTNULLABLE_OBJECT_REF_METHODS (ShapeStructInfo, StructInfo, ShapeStructInfoNode)
 
- Public Member Functions inherited from tvm::relax::StructInfo
 TVM_DEFINE_OBJECT_REF_METHODS (StructInfo, ObjectRef, StructInfoNode)
 

Detailed Description

Managed reference to ShapeStructInfoNode.

See also
ShapeStructInfoNode

Constructor & Destructor Documentation

◆ ShapeStructInfo() [1/2]

tvm::relax::ShapeStructInfo::ShapeStructInfo ( Array< PrimExpr values,
Span  span = Span() 
)

Construction with known symbolic shape patterns.

Parameters
valuesThe symbolic shape values
spanThe span of the AST.

◆ ShapeStructInfo() [2/2]

tvm::relax::ShapeStructInfo::ShapeStructInfo ( int  ndim,
Span  span = Span() 
)

Construction with known unknown symbolic shape patterns.

Parameters
ndimNumber of dimensions – can be kUnknownNDim
spanThe span of the AST.

Member Function Documentation

◆ TVM_DEFINE_NOTNULLABLE_OBJECT_REF_METHODS()

tvm::relax::ShapeStructInfo::TVM_DEFINE_NOTNULLABLE_OBJECT_REF_METHODS ( ShapeStructInfo  ,
StructInfo  ,
ShapeStructInfoNode   
)

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