tvm
Loading...
Searching...
No Matches
Public Member Functions | Static Public Attributes | List of all members
tvm::te::Reduce Class Reference

Managed reference to ReduceNode. More...

#include <operation.h>

Inheritance diagram for tvm::te::Reduce:
Collaboration diagram for tvm::te::Reduce:

Public Member Functions

 Reduce (CommReducer combiner, ffi::Array< PrimExpr > src, ffi::Array< s_tir::IterVar > rdom, ffi::Optional< PrimExpr > condition, int value_index, ffi::Array< PrimExpr > init, Location loc=UnknownLoc())
 
 Reduce (ffi::ObjectPtr< ReduceNode > node)
 
 TVM_FFI_DEFINE_OBJECT_REF_METHODS_NOTNULLABLE (Reduce, PrimExpr, ReduceNode)
 
 TVM_DEFINE_OBJECT_REF_COW_METHOD (ReduceNode)
 
- Public Member Functions inherited from tvm::PrimExpr
TVM_TIRX_DLL PrimExpr (Call call)
 Construct from a call after checking that its result type is PrimType.
 
TVM_TIRX_DLL PrimExpr (int32_t value)
 construct from integer.
 
TVM_TIRX_DLL PrimExpr (float value)
 construct from float.
 
 PrimExpr (ffi::ObjectPtr< ExprNode > node)
 
 TVM_FFI_DEFINE_OBJECT_REF_METHODS_NOTNULLABLE (PrimExpr, TypedExpr< PrimType >, ExprNode)
 
- Public Member Functions inherited from tvm::TypedExpr< PrimType >
PrimType ty () const
 
 TypedExpr (ffi::ObjectPtr< ExprNode > node)
 
 TVM_FFI_DEFINE_OBJECT_REF_METHODS_NOTNULLABLE (TypedExpr, Expr, ExprNode)
 
- Public Member Functions inherited from tvm::Expr
bool operator== (const Expr &other) const =delete
 
bool operator!= (const Expr &other) const =delete
 
bool operator< (const Expr &other) const =delete
 
 Expr (ffi::ObjectPtr< ExprNode > node)
 
 TVM_FFI_DEFINE_OBJECT_REF_METHODS_NOTNULLABLE (Expr, ffi::ObjectRef, ExprNode)
 

Static Public Attributes

static constexpr bool _type_container_is_exact = true
 
- Static Public Attributes inherited from tvm::PrimExpr
static constexpr bool _type_container_is_exact = false
 
- Static Public Attributes inherited from tvm::TypedExpr< PrimType >
static constexpr bool _type_container_is_exact
 

Detailed Description

Managed reference to ReduceNode.

See also
ReduceNode

Constructor & Destructor Documentation

◆ Reduce() [1/2]

tvm::te::Reduce::Reduce ( CommReducer  combiner,
ffi::Array< PrimExpr >  src,
ffi::Array< s_tir::IterVar >  rdom,
ffi::Optional< PrimExpr >  condition,
int  value_index,
ffi::Array< PrimExpr >  init,
Location  loc = UnknownLoc() 
)

◆ Reduce() [2/2]

tvm::te::Reduce::Reduce ( ffi::ObjectPtr< ReduceNode >  node)
inlineexplicit

Member Function Documentation

◆ TVM_DEFINE_OBJECT_REF_COW_METHOD()

tvm::te::Reduce::TVM_DEFINE_OBJECT_REF_COW_METHOD ( ReduceNode  )

◆ TVM_FFI_DEFINE_OBJECT_REF_METHODS_NOTNULLABLE()

tvm::te::Reduce::TVM_FFI_DEFINE_OBJECT_REF_METHODS_NOTNULLABLE ( Reduce  ,
PrimExpr  ,
ReduceNode   
)

Member Data Documentation

◆ _type_container_is_exact

constexpr bool tvm::te::Reduce::_type_container_is_exact = true
staticconstexpr

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