|
tvm
|
A frame that represents if statement. More...
#include <frame.h>
Public Member Functions | |
| TVM_FFI_DECLARE_OBJECT_INFO_FINAL ("script.ir_builder.relax.IfFrame", IfFrameNode, RelaxFrameNode) | |
| void | EnterWithScope () final |
| The method called when entering RAII scope. More... | |
| void | ExitWithScope () final |
| The method called when exiting RAII scope. More... | |
Public Member Functions inherited from tvm::script::ir_builder::relax::RelaxFrameNode | |
| TVM_FFI_DECLARE_OBJECT_INFO ("script.ir_builder.relax.RelaxFrame", RelaxFrameNode, IRBuilderFrameNode) | |
Public Member Functions inherited from tvm::script::ir_builder::IRBuilderFrameNode | |
| TVM_FFI_DECLARE_OBJECT_INFO ("script.ir_builder.IRBuilderFrame", 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 Public Member Functions inherited from tvm::script::ir_builder::relax::RelaxFrameNode | |
| static void | RegisterReflection () |
Static Public Member Functions inherited from tvm::script::ir_builder::IRBuilderFrameNode | |
| static void | RegisterReflection () |
Public Attributes | |
| tvm::relax::Expr | condition |
| The condition of the if statement. More... | |
| ffi::Optional< tvm::relax::Expr > | then_expr |
| The Bindings in the true branch. More... | |
| ffi::Optional< tvm::relax::Expr > | else_expr |
| The Bindings in the false branch. More... | |
| tvm::relax::Var | var |
| The Binding var. More... | |
| ffi::String | var_name |
| The binding var name. More... | |
Public Attributes inherited from tvm::script::ir_builder::IRBuilderFrameNode | |
| std::vector< ffi::TypedFunction< void()> > | callbacks |
| A list of callbacks used when exiting the frame. More... | |
Additional Inherited Members | |
Static Public Attributes inherited from tvm::script::ir_builder::IRBuilderFrameNode | |
| static constexpr const bool | _type_mutable = true |
A frame that represents if statement.
|
finalvirtual |
The method called when entering RAII scope.
Reimplemented from tvm::script::ir_builder::IRBuilderFrameNode.
|
finalvirtual |
The method called when exiting RAII scope.
Reimplemented from tvm::script::ir_builder::IRBuilderFrameNode.
|
inlinestatic |
| tvm::script::ir_builder::relax::IfFrameNode::TVM_FFI_DECLARE_OBJECT_INFO_FINAL | ( | "script.ir_builder.relax.IfFrame" | , |
| IfFrameNode | , | ||
| RelaxFrameNode | |||
| ) |
| tvm::relax::Expr tvm::script::ir_builder::relax::IfFrameNode::condition |
The condition of the if statement.
| ffi::Optional<tvm::relax::Expr> tvm::script::ir_builder::relax::IfFrameNode::else_expr |
The Bindings in the false branch.
| ffi::Optional<tvm::relax::Expr> tvm::script::ir_builder::relax::IfFrameNode::then_expr |
The Bindings in the true branch.
| tvm::relax::Var tvm::script::ir_builder::relax::IfFrameNode::var |
The Binding var.
| ffi::String tvm::script::ir_builder::relax::IfFrameNode::var_name |
The binding var name.