tvm
|
Reference to shape tuple objects. More...
#include <shape_tuple.h>
Public Types | |
using | index_type = ShapeTupleObj::index_type |
The type of shape index element. More... | |
Public Types inherited from tvm::runtime::ObjectRef | |
using | ContainerType = Object |
type indicate the container type. More... | |
Public Member Functions | |
ShapeTuple () | |
Construct an empty shape tuple. More... | |
template<typename IterType > | |
ShapeTuple (IterType begin, IterType end) | |
Constructor from iterator. More... | |
ShapeTuple (std::initializer_list< index_type > shape) | |
constructor from initializer list More... | |
ShapeTuple (std::vector< index_type > shape) | |
Construct a new ShapeTuple object. More... | |
const index_type * | data () const |
Return the data pointer. More... | |
size_t | size () const |
Return the size of the shape tuple. More... | |
index_type | operator[] (size_t idx) const |
Immutably read i-th element from the shape tuple. More... | |
index_type | at (size_t idx) const |
Immutably read i-th element from the shape tuple. More... | |
bool | empty () const |
index_type | front () const |
index_type | back () const |
const index_type * | begin () const |
const index_type * | end () const |
TVM_DEFINE_NOTNULLABLE_OBJECT_REF_METHODS (ShapeTuple, ObjectRef, ShapeTupleObj) | |
Public Member Functions inherited from tvm::runtime::ObjectRef | |
ObjectRef ()=default | |
default constructor More... | |
ObjectRef (ObjectPtr< Object > data) | |
Constructor from existing object ptr. More... | |
bool | same_as (const ObjectRef &other) const |
Comparator. More... | |
bool | operator== (const ObjectRef &other) const |
Comparator. More... | |
bool | operator!= (const ObjectRef &other) const |
Comparator. More... | |
bool | operator< (const ObjectRef &other) const |
Comparator. More... | |
bool | defined () const |
const Object * | get () const |
const Object * | operator-> () const |
bool | unique () const |
int | use_count () const |
template<typename ObjectType , typename = std::enable_if_t<std::is_base_of_v<Object, ObjectType>>> | |
const ObjectType * | as () const |
Try to downcast the internal Object to a raw pointer of a corresponding type. More... | |
template<typename ObjectRefType , typename = std::enable_if_t<std::is_base_of_v<ObjectRef, ObjectRefType>>> | |
Optional< ObjectRefType > | as () const |
Try to downcast the ObjectRef to a Optional<T> of the requested type. More... | |
Additional Inherited Members | |
Static Public Attributes inherited from tvm::runtime::ObjectRef | |
static constexpr bool | _type_is_nullable = true |
Protected Member Functions inherited from tvm::runtime::ObjectRef | |
Object * | get_mutable () const |
Static Protected Member Functions inherited from tvm::runtime::ObjectRef | |
template<typename T > | |
static T | DowncastNoCheck (ObjectRef ref) |
Internal helper function downcast a ref without check. More... | |
static void | FFIClearAfterMove (ObjectRef *ref) |
Clear the object ref data field without DecRef after we successfully moved the field. More... | |
template<typename ObjectType > | |
static ObjectPtr< ObjectType > | GetDataPtr (const ObjectRef &ref) |
Internal helper function get data_ as ObjectPtr of ObjectType. More... | |
Protected Attributes inherited from tvm::runtime::ObjectRef | |
ObjectPtr< Object > | data_ |
Internal pointer that backs the reference. More... | |
Reference to shape tuple objects.
The type of shape index element.
|
inline |
Construct an empty shape tuple.
|
inline |
Constructor from iterator.
begin | begin of iterator |
end | end of iterator |
IterType | The type of iterator |
|
inline |
constructor from initializer list
shape | The initializer list |
|
inline |
Construct a new ShapeTuple object.
shape | The moved/copied std::vector object |
|
inline |
Immutably read i-th element from the shape tuple.
idx | The index |
|
inline |
|
inline |
|
inline |
Return the data pointer.
|
inline |
|
inline |
|
inline |
|
inline |
Immutably read i-th element from the shape tuple.
idx | The index |
|
inline |
Return the size of the shape tuple.
tvm::runtime::ShapeTuple::TVM_DEFINE_NOTNULLABLE_OBJECT_REF_METHODS | ( | ShapeTuple | , |
ObjectRef | , | ||
ShapeTupleObj | |||
) |