tvm
Classes | Functions
tvm::transform Namespace Reference

Classes

class  PassContextNode
 PassContextNode contains the information that a pass can rely on, such as analysis results. More...
 
class  PassContext
 PassContext that is used to configure the pass behavior. More...
 
class  PassInfoNode
 Meta data that will be used to help optimization and analysis. More...
 
class  PassInfo
 Managed reference class for PassInfoNode. More...
 
class  PassNode
 PassNode is the base type of differnt types of optimization passes. It is designed as a pure class and implemented by different pass subclasses at different granularity of Relay/Relax nodes. More...
 
class  Pass
 
class  SequentialNode
 The SequentialNode contains a set of passes that transform Relay/Relax programs from one AST to another semantically equivalent one. More...
 
class  Sequential
 

Functions

Pass CreateModulePass (const runtime::TypedPackedFunc< IRModule(IRModule, PassContext)> &pass_func, int opt_level, String name, Array< runtime::String > required, bool traceable=false)
 
Pass ApplyPassToFunction (Pass pass, String func_name_regex, bool error_if_no_function_matches_regex=false)
 
Pass PrintIR (String header="", bool show_meta_data=false)
 A special trace pass that prints the header and IR to LOG(INFO). More...
 

Function Documentation

◆ ApplyPassToFunction()

Pass tvm::transform::ApplyPassToFunction ( Pass  pass,
String  func_name_regex,
bool  error_if_no_function_matches_regex = false 
)

◆ CreateModulePass()

Pass tvm::transform::CreateModulePass ( const runtime::TypedPackedFunc< IRModule(IRModule, PassContext)> &  pass_func,
int  opt_level,
String  name,
Array< runtime::String required,
bool  traceable = false 
)

◆ PrintIR()

Pass tvm::transform::PrintIR ( String  header = "",
bool  show_meta_data = false 
)

A special trace pass that prints the header and IR to LOG(INFO).

Parameters
headerThe header to be attached to the output.
show_meta_dataWhether should we show meta data.
Returns
The pass.