tvm
Public Member Functions | List of all members
tvm::relax::If Class Reference

#include <expr.h>

Inheritance diagram for tvm::relax::If:
Collaboration diagram for tvm::relax::If:

Public Member Functions

 If (Expr cond, Expr true_branch, Expr false_branch, Span span=Span())
 The constructor. More...
 
 TVM_DEFINE_OBJECT_REF_METHODS (If, Expr, IfNode)
 
 TVM_DEFINE_OBJECT_REF_COW_METHOD (IfNode)
 
- Public Member Functions inherited from tvm::RelaxExpr
 TVM_DEFINE_OBJECT_REF_METHODS (RelaxExpr, BaseExpr, RelaxExprNode)
 
- Public Member Functions inherited from tvm::BaseExpr
 TVM_DEFINE_OBJECT_REF_METHODS (BaseExpr, ObjectRef, BaseExprNode)
 

Constructor & Destructor Documentation

◆ If()

tvm::relax::If::If ( Expr  cond,
Expr  true_branch,
Expr  false_branch,
Span  span = Span() 
)

The constructor.

Parameters
condThe condition of a if node.
true_branchThe fall through branch. If this is not a SeqExpr, it will be wrapped in a SeqExpr, to satisfy the Relax IR requirement that all scopes be contained in a SeqExpr.
false_branchThe branch for execution when condition is false. If this is not a SeqExpr, it will be wrapped in a SeqExpr, to satisfy the Relax IR requirement that all scopes be contained in a SeqExpr.
spanThe source span of the expression.

Member Function Documentation

◆ TVM_DEFINE_OBJECT_REF_COW_METHOD()

tvm::relax::If::TVM_DEFINE_OBJECT_REF_COW_METHOD ( IfNode  )

◆ TVM_DEFINE_OBJECT_REF_METHODS()

tvm::relax::If::TVM_DEFINE_OBJECT_REF_METHODS ( If  ,
Expr  ,
IfNode   
)

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