|
FunctionFrame | tvm::script::ir_builder::relax::Function (const Bool &is_pure, const Bool &is_private) |
| Start a function frame. More...
|
|
tvm::relax::Var | tvm::script::ir_builder::relax::Arg (const String &name, const tvm::relax::StructInfo &struct_info) |
| Add a parameter to the last function frame. More...
|
|
void | tvm::script::ir_builder::relax::FuncName (const String &name) |
| Specify the name of the last function frame. More...
|
|
void | tvm::script::ir_builder::relax::FuncAttrs (Map< String, ObjectRef > attrs) |
| Specify the attrs of the last function frame. More...
|
|
void | tvm::script::ir_builder::relax::FuncRetStructInfo (const tvm::relax::StructInfo &ret_sinfo) |
| Specify the return struct info of the last function frame. More...
|
|
void | tvm::script::ir_builder::relax::FuncRetValue (const tvm::relax::Expr &value) |
| Specify the return value of the last function frame. More...
|
|
BlockFrame | tvm::script::ir_builder::relax::BindingBlock () |
| Start a binding block frame. More...
|
|
BlockFrame | tvm::script::ir_builder::relax::Dataflow () |
| Start a dataflow binding block frame. More...
|
|
void | tvm::script::ir_builder::relax::DataflowBlockOutput (const Array< tvm::relax::Var > &vars) |
| Expose the dataflow block output variables as global ones. More...
|
|
tvm::relax::Var | tvm::script::ir_builder::relax::Emit (const tvm::relax::Expr &value, const Optional< tvm::relax::StructInfo > &annotate_struct_info=NullOpt) |
| Emit a binding to the last binding block frame. More...
|
|
tvm::relax::Var | tvm::script::ir_builder::relax::EmitMatchCast (const tvm::relax::Expr &value, const tvm::relax::StructInfo &struct_info) |
| Emit a match_cast binding to the last binding block frame. More...
|
|
tvm::relax::Var | tvm::script::ir_builder::relax::EmitVarBinding (const tvm::relax::VarBinding &binding) |
| Emit a binding to the last binding block frame. More...
|
|
IfFrame | tvm::script::ir_builder::relax::If (tvm::relax::Expr condition) |
| Create an if statement. More...
|
|
ThenFrame | tvm::script::ir_builder::relax::Then () |
| Create a then. More...
|
|
ElseFrame | tvm::script::ir_builder::relax::Else () |
| Create an else. More...
|
|