ir.h File Reference

#include <tvm/relax/expr.h>
#include <tvm/relax/script/builder/frame.h>
#include <tvm/relax/type.h>
#include <tvm/script/ir_builder/base.h>

Include dependency graph for ir.h:

Go to the source code of this file.

Namespaces
	tvm
	An object that builds and maintains block scope and StmtSref mapping for Dependence analysis.
	tvm::script
	tvm::script::ir_builder
	tvm::script::ir_builder::relax

Functions
FunctionFrame	tvm::script::ir_builder::relax::Function (bool is_pure, bool is_private)
	Start a function frame. More...
tvm::relax::Var	tvm::script::ir_builder::relax::Arg (const ffi::String &name, const tvm::Type &ty)
	Add a parameter to the last function frame. More...
void	tvm::script::ir_builder::relax::FuncName (const ffi::String &name)
	Specify the name of the last function frame. More...
void	tvm::script::ir_builder::relax::FuncAttrs (ffi::Map< ffi::String, Any > attrs)
	Specify the attrs of the last function frame. More...
void	tvm::script::ir_builder::relax::FuncRetType (const tvm::Type &ret_ty)
	Specify the return type 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...
BindingBlockFrame	tvm::script::ir_builder::relax::BindingBlock ()
	Start a binding block frame. More...
BindingBlockFrame	tvm::script::ir_builder::relax::Dataflow ()
	Start a dataflow binding block frame. More...
void	tvm::script::ir_builder::relax::DataflowBlockOutput (const ffi::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 ffi::Optional< tvm::Type > &annotate_ty=std::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::Type &ty)
	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...