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

Managed reference to ShuffleNode. More...

#include <vector_expr.h>

Inheritance diagram for tvm::prim::Shuffle:
Collaboration diagram for tvm::prim::Shuffle:

Public Member Functions

TVM_TIRX_DLL Shuffle (ffi::Array< PrimExpr > vectors, ffi::Array< PrimExpr > indices, Location loc=UnknownLoc())
 
 Shuffle (ffi::ObjectPtr< ShuffleNode > node)
 
 TVM_FFI_DEFINE_OBJECT_REF_METHODS_NOTNULLABLE (Shuffle, PrimExpr, ShuffleNode)
 
 TVM_DEFINE_OBJECT_REF_COW_METHOD (ShuffleNode)
 
- 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 Member Functions

static TVM_TIRX_DLL PrimExpr Concat (ffi::Array< PrimExpr > vectors, Location loc=UnknownLoc())
 
static TVM_TIRX_DLL PrimExpr ExtractElement (PrimExpr vector, int index, Location loc=UnknownLoc())
 

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 ShuffleNode.

See also
ShuffleNode

Constructor & Destructor Documentation

◆ Shuffle() [1/2]

TVM_TIRX_DLL tvm::prim::Shuffle::Shuffle ( ffi::Array< PrimExpr >  vectors,
ffi::Array< PrimExpr >  indices,
Location  loc = UnknownLoc() 
)

◆ Shuffle() [2/2]

tvm::prim::Shuffle::Shuffle ( ffi::ObjectPtr< ShuffleNode >  node)
inlineexplicit

Member Function Documentation

◆ Concat()

static TVM_TIRX_DLL PrimExpr tvm::prim::Shuffle::Concat ( ffi::Array< PrimExpr >  vectors,
Location  loc = UnknownLoc() 
)
static

◆ ExtractElement()

static TVM_TIRX_DLL PrimExpr tvm::prim::Shuffle::ExtractElement ( PrimExpr  vector,
int  index,
Location  loc = UnknownLoc() 
)
static

◆ TVM_DEFINE_OBJECT_REF_COW_METHOD()

tvm::prim::Shuffle::TVM_DEFINE_OBJECT_REF_COW_METHOD ( ShuffleNode  )

◆ TVM_FFI_DEFINE_OBJECT_REF_METHODS_NOTNULLABLE()

tvm::prim::Shuffle::TVM_FFI_DEFINE_OBJECT_REF_METHODS_NOTNULLABLE ( Shuffle  ,
PrimExpr  ,
ShuffleNode   
)

Member Data Documentation

◆ _type_container_is_exact

constexpr bool tvm::prim::Shuffle::_type_container_is_exact = true
staticconstexpr

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