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
	ShapePatternNode (DFPattern pattern)
	ShapePatternNode (ffi::UnsafeInit)
	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, ffi::Object)

Static Public Member Functions
static void	RegisterReflection ()

Public Attributes
DFPattern	pattern
ffi::Array< PrimExpr >	shape

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

Constructor & Destructor Documentation

ShapePatternNode() [1/2]

tvm::relax::ShapePatternNode::ShapePatternNode ( DFPattern  pattern )

inlineexplicit

ShapePatternNode() [2/2]

tvm::relax::ShapePatternNode::ShapePatternNode ( ffi::UnsafeInit  )

inlineexplicit

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:

• include/tvm/relax/dataflow_pattern.h