tvm
|
The base ir_builder frame for frames with SeqExpr i.e. Functions, If branches. More...
#include <frame.h>
Public Member Functions | |
TVM_DECLARE_BASE_OBJECT_INFO (SeqExprFrameNode, RelaxFrameNode) | |
void | EnterWithScope () override |
The method called when entering RAII scope. More... | |
void | ExitWithScope () override |
The method called when exiting RAII scope. More... | |
![]() | |
TVM_DECLARE_BASE_OBJECT_INFO (RelaxFrameNode, IRBuilderFrameNode) | |
![]() | |
TVM_DECLARE_BASE_OBJECT_INFO (IRBuilderFrameNode, runtime::Object) | |
virtual | ~IRBuilderFrameNode ()=default |
Default destructor. More... | |
void | AddCallback (ffi::TypedFunction< void()> callback) |
Add a callback method invoked when exiting the RAII scope. More... | |
Static Public Member Functions | |
static void | RegisterReflection () |
![]() | |
static void | RegisterReflection () |
![]() | |
static void | RegisterReflection () |
Public Attributes | |
Array< tvm::relax::BindingBlock > | binding_blocks |
The binding blocks inside the frame. More... | |
Optional< tvm::relax::Expr > | output |
The frame output expr. std::nullopt when undefined. More... | |
![]() | |
std::vector< ffi::TypedFunction< void()> > | callbacks |
A list of callbacks used when exiting the frame. More... | |
Static Public Attributes | |
static constexpr const char * | _type_key = "script.ir_builder.relax.SeqExprFrame" |
![]() | |
static constexpr const char * | _type_key = "script.ir_builder.relax.RelaxFrame" |
![]() | |
static constexpr const char * | _type_key = "script.ir_builder.IRBuilderFrame" |
The base ir_builder frame for frames with SeqExpr i.e. Functions, If branches.
|
overridevirtual |
The method called when entering RAII scope.
Reimplemented from tvm::script::ir_builder::IRBuilderFrameNode.
Reimplemented in tvm::script::ir_builder::relax::ElseFrameNode, tvm::script::ir_builder::relax::ThenFrameNode, and tvm::script::ir_builder::relax::FunctionFrameNode.
|
overridevirtual |
The method called when exiting RAII scope.
Reimplemented from tvm::script::ir_builder::IRBuilderFrameNode.
Reimplemented in tvm::script::ir_builder::relax::ElseFrameNode, tvm::script::ir_builder::relax::ThenFrameNode, and tvm::script::ir_builder::relax::FunctionFrameNode.
|
inlinestatic |
tvm::script::ir_builder::relax::SeqExprFrameNode::TVM_DECLARE_BASE_OBJECT_INFO | ( | SeqExprFrameNode | , |
RelaxFrameNode | |||
) |
|
staticconstexpr |
Array<tvm::relax::BindingBlock> tvm::script::ir_builder::relax::SeqExprFrameNode::binding_blocks |
The binding blocks inside the frame.
Optional<tvm::relax::Expr> tvm::script::ir_builder::relax::SeqExprFrameNode::output |
The frame output expr. std::nullopt
when undefined.