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_FFI_DECLARE_OBJECT_INFO_FINAL ("relax.transform.PatternCheckContext", PatternCheckContextNode, Object)

Static Public Member Functions
static void	RegisterReflection ()

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

Detailed Description

The input of FusionPattern::check.

Member Function Documentation

RegisterReflection()

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

inlinestatic

TVM_FFI_DECLARE_OBJECT_INFO_FINAL()

tvm::relax::transform::PatternCheckContextNode::TVM_FFI_DECLARE_OBJECT_INFO_FINAL	(	"relax.transform.PatternCheckContext"	,
		PatternCheckContextNode	,
		Object
	)

Member Data Documentation

annotated_expr

ffi::Map<ffi::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

ffi::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

ffi::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

ffi::Map<Var, ffi::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