tvm::relax::DataflowBlockRewriteNode Class Reference

Statement rewriter for relax.DataflowBlock. More...

#include <binding_rewrite.h>

Inheritance diagram for tvm::relax::DataflowBlockRewriteNode:

Collaboration diagram for tvm::relax::DataflowBlockRewriteNode:

Public Member Functions
void	ReplaceAllUses (Var old_var, Var new_var)
	Replace all uses of old_var with new_var. More...
void	Add (Binding binding)
	Insert a Binding statement. More...
void	Add (String var_name, Expr expr, bool is_dfvar=false)
	Insert an expression as VarBinding with variable name. More...
void	Add (Expr expr, bool is_dfvar=false)
	Insert an expression as VarBinding with automatic variable name. More...
void	RemoveUnused (Var unused, bool allow_undef=false)
	Remove the definition statement of an unused variable. More...
void	RemoveAllUnused ()
	Remove the definition statements of all unused variables. More...
DataflowBlock	MutatedDataflowBlock ()
	The rewritten dataflow block. More...
Function	MutatedFunc ()
	The rewritten function. More...
IRModule	MutateIRModule (IRModule irmod)
	The rewritten IRModule. More...
	TVM_DECLARE_FINAL_OBJECT_INFO (DataflowBlockRewriteNode, Object)

Static Public Member Functions
static void	RegisterReflection ()
	Visit attributes. More...

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

Protected Attributes
DataflowBlock	dfb_
	The rewritten dataflow block. More...
Optional< Function >	root_fn_
	The rewritten function. More...
const FunctionNode *	original_fn_ptr_
	Pointer to the original function. More...
Map< Var, Array< Var > >	to_users_
	Map from variable to its users. More...
Array< Var >	fn_outputs_
	Variables required by function outputs. More...

Friends
class	DataflowBlockRewrite

Detailed Description

Statement rewriter for relax.DataflowBlock.

Member Function Documentation

Add() [1/3]

void tvm::relax::DataflowBlockRewriteNode::Add

(

Binding

binding

)

Insert a Binding statement.

Add() [2/3]

void tvm::relax::DataflowBlockRewriteNode::Add ( Expr  expr, bool  is_dfvar = false  )

inline

Insert an expression as VarBinding with automatic variable name.

Add() [3/3]

void tvm::relax::DataflowBlockRewriteNode::Add ( String  var_name, Expr  expr, bool  is_dfvar = false  )

inline

Insert an expression as VarBinding with variable name.

MutatedDataflowBlock()

DataflowBlock tvm::relax::DataflowBlockRewriteNode::MutatedDataflowBlock ( )

inline

The rewritten dataflow block.

MutatedFunc()

Function tvm::relax::DataflowBlockRewriteNode::MutatedFunc ( )

inline

The rewritten function.

MutateIRModule()

IRModule tvm::relax::DataflowBlockRewriteNode::MutateIRModule

(

IRModule

irmod

)

The rewritten IRModule.

RegisterReflection()

static void tvm::relax::DataflowBlockRewriteNode::RegisterReflection ( )

inlinestatic

Visit attributes.

RemoveAllUnused()

void tvm::relax::DataflowBlockRewriteNode::RemoveAllUnused

(

)

Remove the definition statements of all unused variables.

RemoveUnused()

void tvm::relax::DataflowBlockRewriteNode::RemoveUnused	(	Var	unused,
		bool	allow_undef = false
	)

Remove the definition statement of an unused variable.

ReplaceAllUses()

void tvm::relax::DataflowBlockRewriteNode::ReplaceAllUses	(	Var	old_var,
		Var	new_var
	)

Replace all uses of old_var with new_var.

TVM_DECLARE_FINAL_OBJECT_INFO()

tvm::relax::DataflowBlockRewriteNode::TVM_DECLARE_FINAL_OBJECT_INFO	(	DataflowBlockRewriteNode	,
		Object
	)

Friends And Related Function Documentation

DataflowBlockRewrite

friend class DataflowBlockRewrite

friend

Member Data Documentation

_type_key

constexpr const char* tvm::relax::DataflowBlockRewriteNode::_type_key = "relax.DataflowBlockRewrite"

staticconstexpr

dfb_

DataflowBlock tvm::relax::DataflowBlockRewriteNode::dfb_

protected

The rewritten dataflow block.

fn_outputs_

Array<Var> tvm::relax::DataflowBlockRewriteNode::fn_outputs_

protected

Variables required by function outputs.

original_fn_ptr_

const FunctionNode* tvm::relax::DataflowBlockRewriteNode::original_fn_ptr_

protected

Pointer to the original function.

root_fn_

Optional<Function> tvm::relax::DataflowBlockRewriteNode::root_fn_

protected

The rewritten function.

to_users_

Map<Var, Array<Var> > tvm::relax::DataflowBlockRewriteNode::to_users_

protected

Map from variable to its users.

---

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

• include/tvm/relax/binding_rewrite.h