19 #ifndef TVM_SCRIPT_IR_BUILDER_RELAX_IR_H_
20 #define TVM_SCRIPT_IR_BUILDER_RELAX_IR_H_
29 namespace ir_builder {
104 const Optional<tvm::relax::StructInfo>& annotate_struct_info = std::nullopt);
Boolean constant.
Definition: expr.h:577
Managed reference to RelaxExprNode.
Definition: expr.h:446
Managed reference to StructInfoNode.
Definition: expr.h:135
Managed reference to ElseFrameNode.
Definition: frame.h:306
Managed reference to IfFrameNode.
Definition: frame.h:230
Managed reference to ThenFrameNode.
Definition: frame.h:268
tvm::relax::Var EmitMatchCast(const tvm::relax::Expr &value, const tvm::relax::StructInfo &struct_info)
Emit a match_cast binding to the last binding block frame.
void FuncAttrs(Map< String, Any > attrs)
Specify the attrs of the last function frame.
BlockFrame BindingBlock()
Start a binding block frame.
tvm::relax::Var EmitVarBinding(const tvm::relax::VarBinding &binding)
Emit a binding to the last binding block frame.
IfFrame If(tvm::relax::Expr condition)
Create an if statement.
FunctionFrame Function(const Bool &is_pure, const Bool &is_private)
Start a function frame.
ThenFrame Then()
Create a then.
ElseFrame Else()
Create an else.
tvm::relax::Var Arg(const String &name, const tvm::relax::StructInfo &struct_info)
Add a parameter to the last function frame.
void FuncRetStructInfo(const tvm::relax::StructInfo &ret_sinfo)
Specify the return struct info of the last function frame.
void FuncName(const String &name)
Specify the name of the last function frame.
BlockFrame Dataflow()
Start a dataflow binding block frame.
tvm::relax::Var Emit(const tvm::relax::Expr &value, const Optional< tvm::relax::StructInfo > &annotate_struct_info=std::nullopt)
Emit a binding to the last binding block frame.
void FuncRetValue(const tvm::relax::Expr &value)
Specify the return value of the last function frame.
void DataflowBlockOutput(const Array< tvm::relax::Var > &vars)
Expose the dataflow block output variables as global ones.
Performance counters for profiling via the PAPI library.
Definition: analyzer.h:37