tvm
Loading...
Searching...
No Matches
Public Member Functions | List of all members
tvm::s_tir::SBlock Class Reference

Managed reference to SBlockNode. More...

#include <stmt.h>

Inheritance diagram for tvm::s_tir::SBlock:
Collaboration diagram for tvm::s_tir::SBlock:

Public Member Functions

 SBlock (ffi::Array< s_tir::IterVar > iter_vars, ffi::Array< TensorRegion > reads, ffi::Array< TensorRegion > writes, ffi::String name_hint, SeqStmt body, ffi::Optional< SeqStmt > init=std::nullopt, ffi::Array< tirx::TensorVar > alloc_tensors=ffi::Array< tirx::TensorVar >(), ffi::Array< MatchTensorRegion > match_tensors=ffi::Array< MatchTensorRegion >(), ffi::Map< ffi::String, ffi::Any > annotations=ffi::Map< ffi::String, ffi::Any >(), Location loc=UnknownLoc())
 
 SBlock (ffi::String name_hint, SeqStmt body, ffi::Array< tirx::TensorVar > alloc_tensors=ffi::Array< tirx::TensorVar >(), Location loc=UnknownLoc())
 
 SBlock (ffi::ObjectPtr< SBlockNode > node)
 
 TVM_FFI_DEFINE_OBJECT_REF_METHODS_NOTNULLABLE (SBlock, Stmt, SBlockNode)
 
 TVM_DEFINE_OBJECT_REF_COW_METHOD (SBlockNode)
 
- Public Member Functions inherited from tvm::Stmt
 Stmt (ffi::ObjectPtr< StmtNode > node)
 
 TVM_FFI_DEFINE_OBJECT_REF_METHODS_NOTNULLABLE (Stmt, ffi::ObjectRef, StmtNode)
 

Detailed Description

Managed reference to SBlockNode.

See also
SBlockNode

Constructor & Destructor Documentation

◆ SBlock() [1/3]

tvm::s_tir::SBlock::SBlock ( ffi::Array< s_tir::IterVar >  iter_vars,
ffi::Array< TensorRegion >  reads,
ffi::Array< TensorRegion >  writes,
ffi::String  name_hint,
SeqStmt  body,
ffi::Optional< SeqStmt >  init = std::nullopt,
ffi::Array< tirx::TensorVar >  alloc_tensors = ffi::Array< tirx::TensorVar >(),
ffi::Array< MatchTensorRegion >  match_tensors = ffi::Array< MatchTensorRegion >(),
ffi::Map< ffi::String, ffi::Any >  annotations = ffi::Map< ffi::String, ffi::Any >(),
Location  loc = UnknownLoc() 
)
explicit

◆ SBlock() [2/3]

tvm::s_tir::SBlock::SBlock ( ffi::String  name_hint,
SeqStmt  body,
ffi::Array< tirx::TensorVar >  alloc_tensors = ffi::Array< tirx::TensorVar >(),
Location  loc = UnknownLoc() 
)
explicit

◆ SBlock() [3/3]

tvm::s_tir::SBlock::SBlock ( ffi::ObjectPtr< SBlockNode >  node)
inlineexplicit

Member Function Documentation

◆ TVM_DEFINE_OBJECT_REF_COW_METHOD()

tvm::s_tir::SBlock::TVM_DEFINE_OBJECT_REF_COW_METHOD ( SBlockNode  )

◆ TVM_FFI_DEFINE_OBJECT_REF_METHODS_NOTNULLABLE()

tvm::s_tir::SBlock::TVM_FFI_DEFINE_OBJECT_REF_METHODS_NOTNULLABLE ( SBlock  ,
Stmt  ,
SBlockNode   
)

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