The ir_builder frame for the relax function.
More...
#include <frame.h>
|
static constexpr const char * | _type_key = "script.ir_builder.relax.FunctionFrame" |
|
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 ir_builder frame for the relax function.
◆ EnterWithScope()
void tvm::script::ir_builder::relax::FunctionFrameNode::EnterWithScope |
( |
| ) |
|
|
finalvirtual |
◆ ExitWithScope()
void tvm::script::ir_builder::relax::FunctionFrameNode::ExitWithScope |
( |
| ) |
|
|
finalvirtual |
◆ RegisterReflection()
static void tvm::script::ir_builder::relax::FunctionFrameNode::RegisterReflection |
( |
| ) |
|
|
inlinestatic |
◆ TVM_DECLARE_FINAL_OBJECT_INFO()
◆ _type_key
constexpr const char* tvm::script::ir_builder::relax::FunctionFrameNode::_type_key = "script.ir_builder.relax.FunctionFrame" |
|
staticconstexpr |
◆ attrs
Map<String, Any> tvm::script::ir_builder::relax::FunctionFrameNode::attrs |
◆ block_builder
The block builder to create Relax function.
◆ is_private
Optional<Bool> tvm::script::ir_builder::relax::FunctionFrameNode::is_private |
Whether the function is annotated as private.
◆ is_pure
Optional<Bool> tvm::script::ir_builder::relax::FunctionFrameNode::is_pure |
Whether the function is annotated as pure.
◆ name
Optional<String> tvm::script::ir_builder::relax::FunctionFrameNode::name |
The function name.
- Note
- The name will not be specified in constructor, so it is "Optional", However, we must specify the name by
R.func_name
before exit this frame.
◆ params
Array<tvm::relax::Var> tvm::script::ir_builder::relax::FunctionFrameNode::params |
◆ ret_struct_info
The function return struct info.
- Note
- Usually the function return type can be deduced by the function body. But we can use this field to specify a more "accurate" return type. i.e. If the
ret_struct_info
is None, try to use the deduced type from body If the ret_struct_info
is not None, we can still take body.struct_info if we ret_struct_info is base of body.struct_info. If not, we will take the specified ret_struct_info
.
The documentation for this class was generated from the following file:
- include/tvm/script/ir_builder/relax/frame.h