tvm::relax::transform::PatternCheckContextNode Class Reference

The input of FusionPattern::check. More...

#include <transform.h>

Inheritance diagram for tvm::relax::transform::PatternCheckContextNode:

Collaboration diagram for tvm::relax::transform::PatternCheckContextNode:

Public Member Functions
	TVM_DECLARE_FINAL_OBJECT_INFO (PatternCheckContextNode, Object)

Static Public Member Functions
static void	RegisterReflection ()

Public Attributes
Expr	matched_expr
	The expression that's matched with the FusionPattern::pattern. More...
Map< String, Expr >	annotated_expr
	A map which contains all expressions matched by the sub patterns in FusionPattern::annotation_patterns. More...
Map< Var, Expr >	matched_bindings
	Map from variable to its value. It contains variables from bindings that is being fused by FuseOpsByPattern. More...
Map< Var, Array< Var > >	var_usages
	A map mapping variable definitions to a set of uses. It has all variables used in the function. More...
Map< Expr, Var >	value_to_bound_var
	Map from value to its bound variable. It doesn't have variables after the matched expression. More...

Static Public Attributes
static constexpr const char *	_type_key = "relax.transform.PatternCheckContext"

Detailed Description

The input of FusionPattern::check.

Member Function Documentation

RegisterReflection()

static void tvm::relax::transform::PatternCheckContextNode::RegisterReflection ( )

inlinestatic

TVM_DECLARE_FINAL_OBJECT_INFO()

tvm::relax::transform::PatternCheckContextNode::TVM_DECLARE_FINAL_OBJECT_INFO	(	PatternCheckContextNode	,
		Object
	)

Member Data Documentation

_type_key

constexpr const char* tvm::relax::transform::PatternCheckContextNode::_type_key = "relax.transform.PatternCheckContext"

staticconstexpr

annotated_expr

Map<String, Expr> tvm::relax::transform::PatternCheckContextNode::annotated_expr

A map which contains all expressions matched by the sub patterns in FusionPattern::annotation_patterns.

matched_bindings

Map<Var, Expr> tvm::relax::transform::PatternCheckContextNode::matched_bindings

Map from variable to its value. It contains variables from bindings that is being fused by FuseOpsByPattern.

matched_expr

Expr tvm::relax::transform::PatternCheckContextNode::matched_expr

The expression that's matched with the FusionPattern::pattern.

value_to_bound_var

Map<Expr, Var> tvm::relax::transform::PatternCheckContextNode::value_to_bound_var

Map from value to its bound variable. It doesn't have variables after the matched expression.

var_usages

Map<Var, Array<Var> > tvm::relax::transform::PatternCheckContextNode::var_usages

A map mapping variable definitions to a set of uses. It has all variables used in the function.

---

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

• include/tvm/relax/transform.h