tvm
Public Member Functions | Static Public Member Functions | Public Attributes | Static Public Attributes | List of all members
tvm::tir::ForNode Class Reference

A for loop, with possible type annotations. More...

#include <stmt.h>

Inheritance diagram for tvm::tir::ForNode:
Collaboration diagram for tvm::tir::ForNode:

Public Member Functions

 TVM_DECLARE_FINAL_OBJECT_INFO (ForNode, StmtNode)
 
- Public Member Functions inherited from tvm::tir::StmtNode
 StmtNode ()=default
 
 StmtNode (Span span)
 
 TVM_OBJECT_ENABLE_SCRIPT_PRINTER ()
 
 TVM_DECLARE_BASE_OBJECT_INFO (StmtNode, Object)
 

Static Public Member Functions

static void RegisterReflection ()
 
- Static Public Member Functions inherited from tvm::tir::StmtNode
static void RegisterReflection ()
 

Public Attributes

Var loop_var
 The loop variable. More...
 
PrimExpr min
 The minimum value of iteration. More...
 
PrimExpr extent
 The extent of the iteration. More...
 
ForKind kind
 The kind of the for loop. More...
 
Stmt body
 The body of the for loop. More...
 
Optional< IterVarthread_binding
 Only valid when kind == ForKind::kThreadBinding The context thread that this loop variable bounds to. More...
 
Map< String, ffi::Any > annotations
 Additional annotations about the loop. More...
 
- Public Attributes inherited from tvm::tir::StmtNode
Span span
 Span that points to the original source code. Reserved debug information. More...
 

Static Public Attributes

static constexpr const char * _type_key = "tir.For"
 
- Static Public Attributes inherited from tvm::tir::StmtNode
static constexpr const char * _type_key = "tir.Stmt"
 
static constexpr TVMFFISEqHashKind _type_s_eq_hash_kind = kTVMFFISEqHashKindTreeNode
 
static constexpr const uint32_t _type_child_slots = 15
 

Detailed Description

A for loop, with possible type annotations.

// body
}
PrimExpr min
The minimum value of iteration.
Definition: stmt.h:751
Var loop_var
The loop variable.
Definition: stmt.h:749
PrimExpr extent
The extent of the iteration.
Definition: stmt.h:753

Member Function Documentation

◆ RegisterReflection()

static void tvm::tir::ForNode::RegisterReflection ( )
inlinestatic

◆ TVM_DECLARE_FINAL_OBJECT_INFO()

tvm::tir::ForNode::TVM_DECLARE_FINAL_OBJECT_INFO ( ForNode  ,
StmtNode   
)

Member Data Documentation

◆ _type_key

constexpr const char* tvm::tir::ForNode::_type_key = "tir.For"
staticconstexpr

◆ annotations

Map<String, ffi::Any> tvm::tir::ForNode::annotations

Additional annotations about the loop.

These annotations can be used as auxiliary hint to future transformations. An annotation should not change the control flow semantics of the loop and can be ignored in most passes.

◆ body

Stmt tvm::tir::ForNode::body

The body of the for loop.

◆ extent

PrimExpr tvm::tir::ForNode::extent

The extent of the iteration.

◆ kind

ForKind tvm::tir::ForNode::kind

The kind of the for loop.

◆ loop_var

Var tvm::tir::ForNode::loop_var

The loop variable.

◆ min

PrimExpr tvm::tir::ForNode::min

The minimum value of iteration.

◆ thread_binding

Optional<IterVar> tvm::tir::ForNode::thread_binding

Only valid when kind == ForKind::kThreadBinding The context thread that this loop variable bounds to.


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