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_DECLARE_FINAL_OBJECT_INFO (ShapePatternNode, DFPatternNode)
Public Member Functions inherited from tvm::relax::DFPatternNode
	TVM_DECLARE_BASE_OBJECT_INFO (DFPatternNode, Object)

Static Public Member Functions
static void	RegisterReflection ()

Public Attributes
DFPattern	pattern
Array< PrimExpr >	shape

Static Public Attributes
static constexpr const char *	_type_key = "relax.dpl.ShapePattern"
Static Public Attributes inherited from tvm::relax::DFPatternNode
static constexpr const char *	_type_key = "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_DECLARE_FINAL_OBJECT_INFO()

tvm::relax::ShapePatternNode::TVM_DECLARE_FINAL_OBJECT_INFO	(	ShapePatternNode	,
		DFPatternNode
	)

Member Data Documentation

_type_key

constexpr const char* tvm::relax::ShapePatternNode::_type_key = "relax.dpl.ShapePattern"

staticconstexpr

pattern

DFPattern tvm::relax::ShapePatternNode::pattern

The root pattern to match

shape

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