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

A pattern that asserting a root pattern has a certain shape. More...

#include <dataflow_pattern.h>

Inheritance diagram for tvm::relax::ShapePatternNode:
Collaboration diagram for tvm::relax::ShapePatternNode:

Public Member Functions

 TVM_FFI_DECLARE_OBJECT_INFO_FINAL ("relax.dpl.ShapePattern", ShapePatternNode, DFPatternNode)
 
- Public Member Functions inherited from tvm::relax::DFPatternNode
 TVM_FFI_DECLARE_OBJECT_INFO ("relax.dpl.DFPattern", DFPatternNode, Object)
 

Static Public Member Functions

static void RegisterReflection ()
 

Public Attributes

DFPattern pattern
 
ffi::Array< PrimExprshape
 

Additional Inherited Members

- Static Public Attributes inherited from tvm::relax::DFPatternNode
static constexpr const uint32_t _type_child_slots = 21
 

Detailed Description

A pattern that asserting a root pattern has a certain shape.

See also
ShapePattern

Member Function Documentation

◆ RegisterReflection()

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

◆ TVM_FFI_DECLARE_OBJECT_INFO_FINAL()

tvm::relax::ShapePatternNode::TVM_FFI_DECLARE_OBJECT_INFO_FINAL ( "relax.dpl.ShapePattern"  ,
ShapePatternNode  ,
DFPatternNode   
)

Member Data Documentation

◆ pattern

DFPattern tvm::relax::ShapePatternNode::pattern

The root pattern to match

◆ shape

ffi::Array<PrimExpr> tvm::relax::ShapePatternNode::shape

The shape to match


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