tvm
Public Member Functions | Static Public Member Functions | Public Attributes | Static Public Attributes | List of all members
tvm::script::ir_builder::relax::IfFrameNode Class Reference

A frame that represents if statement. More...

#include <frame.h>

Inheritance diagram for tvm::script::ir_builder::relax::IfFrameNode:
Collaboration diagram for tvm::script::ir_builder::relax::IfFrameNode:

Public Member Functions

 TVM_DECLARE_FINAL_OBJECT_INFO (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_DECLARE_BASE_OBJECT_INFO (RelaxFrameNode, IRBuilderFrameNode)
 
- Public Member Functions inherited from tvm::script::ir_builder::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 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...
 
Optional< tvm::relax::Exprthen_expr
 The Bindings in the true branch. More...
 
Optional< tvm::relax::Exprelse_expr
 The Bindings in the false branch. More...
 
tvm::relax::Var var
 The Binding var. More...
 
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...
 

Static Public Attributes

static constexpr const char * _type_key = "script.ir_builder.relax.IfFrame"
 
- Static Public Attributes inherited from tvm::script::ir_builder::relax::RelaxFrameNode
static constexpr const char * _type_key = "script.ir_builder.relax.RelaxFrame"
 
- Static Public Attributes inherited from tvm::script::ir_builder::IRBuilderFrameNode
static constexpr const char * _type_key = "script.ir_builder.IRBuilderFrame"
 

Detailed Description

A frame that represents if statement.

See also
IfFrame

Member Function Documentation

◆ EnterWithScope()

void tvm::script::ir_builder::relax::IfFrameNode::EnterWithScope ( )
finalvirtual

The method called when entering RAII scope.

See also
tvm::support::With

Reimplemented from tvm::script::ir_builder::IRBuilderFrameNode.

◆ ExitWithScope()

void tvm::script::ir_builder::relax::IfFrameNode::ExitWithScope ( )
finalvirtual

The method called when exiting RAII scope.

See also
tvm::support::With

Reimplemented from tvm::script::ir_builder::IRBuilderFrameNode.

◆ RegisterReflection()

static void tvm::script::ir_builder::relax::IfFrameNode::RegisterReflection ( )
inlinestatic

◆ TVM_DECLARE_FINAL_OBJECT_INFO()

tvm::script::ir_builder::relax::IfFrameNode::TVM_DECLARE_FINAL_OBJECT_INFO ( IfFrameNode  ,
RelaxFrameNode   
)

Member Data Documentation

◆ _type_key

constexpr const char* tvm::script::ir_builder::relax::IfFrameNode::_type_key = "script.ir_builder.relax.IfFrame"
staticconstexpr

◆ condition

tvm::relax::Expr tvm::script::ir_builder::relax::IfFrameNode::condition

The condition of the if statement.

◆ else_expr

Optional<tvm::relax::Expr> tvm::script::ir_builder::relax::IfFrameNode::else_expr

The Bindings in the false branch.

◆ then_expr

Optional<tvm::relax::Expr> tvm::script::ir_builder::relax::IfFrameNode::then_expr

The Bindings in the true branch.

◆ var

tvm::relax::Var tvm::script::ir_builder::relax::IfFrameNode::var

The Binding var.

◆ var_name

String tvm::script::ir_builder::relax::IfFrameNode::var_name

The binding var name.


The documentation for this class was generated from the following file: