|
tvm
|
The base ir_builder frame for frames with SeqExpr i.e. Functions, IfExpr branches. More...
#include <frame.h>
Public Member Functions | |
| TVM_FFI_DECLARE_OBJECT_INFO ("script.ir_builder.relax.SeqExprFrame", SeqExprFrameObj, RelaxFrameObj) | |
| void | EnterWithScope () override |
| The method called when entering RAII scope. | |
| void | ExitWithScope () override |
| The method called when exiting RAII scope. | |
Public Member Functions inherited from tvm::script::ir_builder::relax::RelaxFrameObj | |
| TVM_FFI_DECLARE_OBJECT_INFO ("script.ir_builder.relax.RelaxFrame", RelaxFrameObj, IRBuilderFrameObj) | |
Public Member Functions inherited from tvm::script::ir_builder::IRBuilderFrameObj | |
| IRBuilderFrameObj () | |
| Capture the source-call context when this frame is constructed. | |
| TVM_FFI_DECLARE_OBJECT_INFO ("script.ir_builder.IRBuilderFrame", IRBuilderFrameObj, ffi::Object) | |
| virtual | ~IRBuilderFrameObj ()=default |
| Default destructor. | |
| void | AddCallback (ffi::TypedFunction< void()> callback) |
| Add a callback method invoked when exiting the RAII scope. | |
Static Public Member Functions | |
| static void | RegisterReflection () |
Static Public Member Functions inherited from tvm::script::ir_builder::relax::RelaxFrameObj | |
| static void | RegisterReflection () |
Static Public Member Functions inherited from tvm::script::ir_builder::IRBuilderFrameObj | |
| static void | RegisterReflection () |
Public Attributes | |
| ffi::Array< tvm::relax::BindingBlock > | binding_blocks |
| The binding blocks inside the frame. | |
| ffi::Optional< tvm::relax::Expr > | output |
The frame output expr. std::nullopt when undefined. | |
Public Attributes inherited from tvm::script::ir_builder::IRBuilderFrameObj | |
| std::vector< ffi::TypedFunction< void()> > | callbacks |
| A list of callbacks used when exiting the frame. | |
| Location | loc = UnknownLoc() |
| Source context retained until this frame constructs its result. | |
Additional Inherited Members | |
Static Public Attributes inherited from tvm::script::ir_builder::IRBuilderFrameObj | |
| static constexpr const bool | _type_mutable = true |
The base ir_builder frame for frames with SeqExpr i.e. Functions, IfExpr branches.
|
overridevirtual |
The method called when entering RAII scope.
Reimplemented from tvm::script::ir_builder::IRBuilderFrameObj.
Reimplemented in tvm::script::ir_builder::relax::ThenFrameObj.
|
overridevirtual |
The method called when exiting RAII scope.
Reimplemented from tvm::script::ir_builder::IRBuilderFrameObj.
Reimplemented in tvm::script::ir_builder::relax::ThenFrameObj.
|
inlinestatic |
| tvm::script::ir_builder::relax::SeqExprFrameObj::TVM_FFI_DECLARE_OBJECT_INFO | ( | "script.ir_builder.relax.SeqExprFrame" | , |
| SeqExprFrameObj | , | ||
| RelaxFrameObj | |||
| ) |
| ffi::Array<tvm::relax::BindingBlock> tvm::script::ir_builder::relax::SeqExprFrameObj::binding_blocks |
The binding blocks inside the frame.
| ffi::Optional<tvm::relax::Expr> tvm::script::ir_builder::relax::SeqExprFrameObj::output |
The frame output expr. std::nullopt when undefined.