tvm
|
Managed reference to ShapeStructInfoNode. More...
#include <struct_info.h>
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) | |
![]() | |
TVM_DEFINE_OBJECT_REF_METHODS (StructInfo, ObjectRef, StructInfoNode) | |
Managed reference to ShapeStructInfoNode.
Construction with known symbolic shape patterns.
values | The symbolic shape values |
span | The span of the AST. |
Construction with known unknown symbolic shape patterns.
ndim | Number of dimensions – can be kUnknownNDim |
span | The span of the AST. |
tvm::relax::ShapeStructInfo::TVM_DEFINE_NOTNULLABLE_OBJECT_REF_METHODS | ( | ShapeStructInfo | , |
StructInfo | , | ||
ShapeStructInfoNode | |||
) |