tvm
Public Member Functions | Static Public Member Functions | Public Attributes | List of all members
tvm::relax::ShapeStructInfoNode Class Reference

StructInfo of shape value. More...

#include <struct_info.h>

Inheritance diagram for tvm::relax::ShapeStructInfoNode:
Collaboration diagram for tvm::relax::ShapeStructInfoNode:

Public Member Functions

bool IsUnknownNdim () const
 
 TVM_FFI_DECLARE_OBJECT_INFO_FINAL ("relax.ShapeStructInfo", ShapeStructInfoNode, StructInfoNode)
 
- Public Member Functions inherited from tvm::relax::StructInfoNode
 TVM_FFI_DECLARE_OBJECT_INFO ("ir.StructInfo", StructInfoNode, Object)
 

Static Public Member Functions

static void RegisterReflection ()
 
- Static Public Member Functions inherited from tvm::relax::StructInfoNode
static void RegisterReflection ()
 

Public Attributes

ffi::Optional< ffi::Array< PrimExpr > > values
 optionally stores the symbolic value patterns of the shape More...
 
int ndim
 The number of dimension of the shape, can be unknown. More...
 
- Public Attributes inherited from tvm::relax::StructInfoNode
Span span
 Span that points to the original source code. Reserved debug information. More...
 

Additional Inherited Members

- Static Public Attributes inherited from tvm::relax::StructInfoNode
static constexpr TVMFFISEqHashKind _type_s_eq_hash_kind = kTVMFFISEqHashKindTreeNode
 
static constexpr const uint32_t _type_child_slots = 7
 

Detailed Description

StructInfo of shape value.

Member Function Documentation

◆ IsUnknownNdim()

bool tvm::relax::ShapeStructInfoNode::IsUnknownNdim ( ) const
inline
Returns
Whether the struct info contains unknown ndim.

◆ RegisterReflection()

static void tvm::relax::ShapeStructInfoNode::RegisterReflection ( )
inlinestatic

◆ TVM_FFI_DECLARE_OBJECT_INFO_FINAL()

tvm::relax::ShapeStructInfoNode::TVM_FFI_DECLARE_OBJECT_INFO_FINAL ( "relax.ShapeStructInfo"  ,
ShapeStructInfoNode  ,
StructInfoNode   
)

Member Data Documentation

◆ ndim

int tvm::relax::ShapeStructInfoNode::ndim

The number of dimension of the shape, can be unknown.

See also
kUnknownNDim

◆ values

ffi::Optional<ffi::Array<PrimExpr> > tvm::relax::ShapeStructInfoNode::values

optionally stores the symbolic value patterns of the shape


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