ir.h File Reference

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

Include dependency graph for ir.h:

Go to the source code of this file.

Namespaces
	tvm
	runtime implementation for LibTorch/TorchScript.
	tvm::script
	tvm::script::ir_builder
	tvm::script::ir_builder::relax

Functions
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...