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

#include <transform.h>

Inheritance diagram for tvm::relax::transform::PatternCheckContextObj:

Collaboration diagram for tvm::relax::transform::PatternCheckContextObj:

Public Member Functions
	PatternCheckContextObj (Expr matched_expr)

	PatternCheckContextObj (ffi::UnsafeInit)

	TVM_FFI_DECLARE_OBJECT_INFO_FINAL ("relax.transform.PatternCheckContext", PatternCheckContextObj, ffi::Object)

Static Public Member Functions
static void	RegisterReflection ()

Public Attributes
Expr	matched_expr
	The expression that's matched with the FusionPattern::pattern.

ffi::Map< ffi::String, Expr >	annotated_expr
	A map which contains all expressions matched by the sub patterns in FusionPattern::annotation_patterns.

ffi::Map< Var, Expr >	matched_bindings
	Map from variable to its value. It contains variables from bindings that is being fused by FuseOpsByPattern.

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.

ffi::Map< Expr, Var >	value_to_bound_var
	Map from value to its bound variable. It doesn't have variables after the matched expression.

Detailed Description

The input of FusionPattern::check.

Constructor & Destructor Documentation

◆ PatternCheckContextObj() [1/2]

tvm::relax::transform::PatternCheckContextObj::PatternCheckContextObj ( Expr matched_expr )

inlineexplicit

◆ PatternCheckContextObj() [2/2]

tvm::relax::transform::PatternCheckContextObj::PatternCheckContextObj ( ffi::UnsafeInit )

inlineexplicit

Member Function Documentation

◆ RegisterReflection()

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

inlinestatic

◆ TVM_FFI_DECLARE_OBJECT_INFO_FINAL()

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

Member Data Documentation

◆ annotated_expr

ffi::Map<ffi::String, Expr> tvm::relax::transform::PatternCheckContextObj::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::PatternCheckContextObj::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::PatternCheckContextObj::matched_expr

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

◆ value_to_bound_var

ffi::Map<Expr, Var> tvm::relax::transform::PatternCheckContextObj::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::PatternCheckContextObj::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

Public Member Functions

Static Public Member Functions

Public Attributes