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
namespace	tvm
	An object that builds and maintains block scope and StmtSref mapping for Dependence analysis.
namespace	tvm::script
namespace	tvm::script::ir_builder
namespace	tvm::script::ir_builder::relax

Functions
FunctionFrame	tvm::script::ir_builder::relax::Function (bool is_pure, bool is_private)
	Start a function frame.
tvm::Var	tvm::script::ir_builder::relax::Arg (const ffi::String &name, const tvm::Type &ty)
	Add a parameter to the last function frame.
void	tvm::script::ir_builder::relax::FuncName (const ffi::String &name)
	Specify the name of the last function frame.
void	tvm::script::ir_builder::relax::FuncAttrs (ffi::Map< ffi::String, Any > attrs)
	Specify the attrs of the last function frame.
void	tvm::script::ir_builder::relax::FuncRetType (const tvm::Type &ret_ty)
	Specify the return type of the last function frame.
void	tvm::script::ir_builder::relax::FuncRetValue (const tvm::relax::Expr &value)
	Specify the return value of the last function frame.
BindingBlockFrame	tvm::script::ir_builder::relax::BindingBlock ()
	Start a binding block frame.
BindingBlockFrame	tvm::script::ir_builder::relax::Dataflow ()
	Start a dataflow binding block frame.
void	tvm::script::ir_builder::relax::DataflowBlockOutput (const ffi::Array< tvm::Var > &vars)
	Expose the dataflow block output variables as global ones.
tvm::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.
tvm::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.
tvm::Var	tvm::script::ir_builder::relax::EmitVarBinding (const tvm::relax::VarBinding &binding)
	Emit a binding to the last binding block frame.
IfFrame	tvm::script::ir_builder::relax::If (tvm::relax::Expr condition)
	Create an if statement.
ThenFrame	tvm::script::ir_builder::relax::Then ()
	Create a then.
ElseFrame	tvm::script::ir_builder::relax::Else ()
	Create an else.