ir.h File Reference

#include <tvm/relax/expr.h>
#include <tvm/relax/script/ir_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
ffi::Optional< tvm::relax::VDevice >	tvm::script::ir_builder::relax::LookupVDevice (ffi::String target_kind, int device_index)
	Find a module virtual device by target kind and index.
FunctionFrame	tvm::script::ir_builder::relax::Function (bool is_pure, bool is_private)
	Start a function frame.
FunctionFrame	tvm::script::ir_builder::relax::DeclFunction (bool is_pure, bool is_private, bool local)
	Start a bodyless declaration using the normal signature operations.
FunctionFrame	tvm::script::ir_builder::relax::LocalFunction (bool is_pure, const tvm::Var &reference)
	Define a local function under its cached declaration identity.
tvm::Var	tvm::script::ir_builder::relax::ArgVar (const ffi::String &name, const tvm::Var &var)
	Add a cached parameter without changing its identity.
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.
tvm::Var	tvm::script::ir_builder::relax::EmitWithLoc (const tvm::relax::Expr &value, const ffi::Optional< tvm::Type > &annotate_ty, const Location &name_loc, const Location &loc=UnknownLoc())
	Emit a binding with separate statement and variable-name ranges.
tvm::Var	tvm::script::ir_builder::relax::EmitMatchCastWithLoc (const tvm::relax::Expr &value, const tvm::Type &ty, const Location &name_loc, const Location &loc=UnknownLoc())
	Emit a match cast with separate statement and variable-name ranges.
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.