|
mxnet
|
Namespaces | |
| detail | |
Classes | |
| class | ADT |
| reference to algebraic data type objects. More... | |
| class | ADTBuilder |
| A builder class that helps to incrementally build ADT. More... | |
| class | ADTObj |
| An object representing a structure or enumeration. More... | |
| struct | array_type_info |
| The type trait indicates subclass of TVM's NDArray. For irrelavant classes, code = -1. For TVM NDArray itself, code = 0. All subclasses of NDArray should override code > 0. More... | |
| class | EllipsisObj |
| Ellipsis. More... | |
| struct | extension_type_info |
| Type traits to mark if a class is tvm extension type. More... | |
| class | InplaceArrayBase |
| Base template for classes with array like memory layout. More... | |
| class | Integer |
| class | IntegerObj |
| class | MXNetArgs |
| Arguments into TVM functions. More... | |
| class | MXNetArgsSetter |
| class | MXNetArgValue |
| A single argument value to PackedFunc. Containing both type_code and MXNetValue. More... | |
| class | MXNetDataType |
| Runtime primitive data type. More... | |
| class | MXNetPODValue_ |
| Internal base class to handle conversion to POD values. More... | |
| class | MXNetRetValue |
| Return Value container, Unlike MXNetArgValue, which only holds reference and do not delete the underlying container during destruction. More... | |
| class | ObjAllocatorBase |
| Base class of object allocators that implements make. Use curiously recurring template pattern. More... | |
| class | Object |
| base class of all object containers. More... | |
| struct | ObjectEqual |
| ObjectRef equal functor. More... | |
| struct | ObjectHash |
| ObjectRef hash functor. More... | |
| class | ObjectPtr |
| A custom smart pointer for Object. More... | |
| class | ObjectRef |
| Base class of all object reference. More... | |
| class | PackedFunc |
| Packed function is a type-erased function. The arguments are passed by packed format. More... | |
| class | Registry |
| Registry for global function. More... | |
| class | SimpleObjAllocator |
| class | Slice |
| class | SliceObj |
| Slice. More... | |
| class | TypedPackedFunc |
| Please refer to TypedPackedFunc<R(Args..)>. More... | |
| class | TypedPackedFunc< R(Args...)> |
| A PackedFunc wrapper to provide typed function signature. It is backed by a PackedFunc internally. More... | |
Enumerations | |
| enum | TypeIndex { kRoot = 0, kMXNetTensor = 1, kMXNetClosure = 2, kMXNetADT = 3, kRuntimeModule = 4, kEllipsis = 5, kSlice = 6, kInteger = 7, kStaticIndexEnd, kDynamic = kStaticIndexEnd } |
| list of the type index. More... | |
Functions | |
| ObjectRef | CreateEllipsis () |
| int64_t | SliceNoneValue () |
| template<typename T , typename... Args> | |
| ObjectPtr< T > | make_object (Args &&...args) |
| Allocate an object using default allocator. More... | |
| template<typename ArrayType , typename ElemType , typename... Args> | |
| ObjectPtr< ArrayType > | make_inplace_array_object (size_t num_elems, Args &&...args) |
| template<typename RefType , typename ObjectType > | |
| RefType | GetRef (const ObjectType *ptr) |
| Get a reference type from a raw object ptr type. More... | |
| template<typename SubRef , typename BaseRef > | |
| SubRef | Downcast (BaseRef ref) |
| Downcast a base reference type to a more specific type. More... | |
| template<typename BaseType , typename ObjectType > | |
| ObjectPtr< BaseType > | GetObjectPtr (ObjectType *ptr) |
| Get an object ptr type from a raw object ptr. More... | |
| template<typename RefType , typename ObjType > | |
| RefType | GetRef (const ObjType *ptr) |
| template<typename BaseType , typename ObjType > | |
| ObjectPtr< BaseType > | GetObjectPtr (ObjType *ptr) |
| DLDataType | String2DLDataType (std::string s) |
| convert a string to TVM type. More... | |
| const char * | TypeCode2Str (int type_code) |
| Convert type code to its name. More... | |
| int | String2MXNetTypeWithBool (const std::string &s) |
| int | String2MXNetType (const std::string &s) |
| std::ostream & | operator<< (std::ostream &os, DLDataType t) |
| std::ostream & | operator<< (std::ostream &os, const MXNetDataType &dtype) |
|
inline |
|
inline |
Downcast a base reference type to a more specific type.
| ref | The inptut reference |
| SubRef | The target specific reference type. |
| BaseRef | the current reference type. |
|
inline |
Get an object ptr type from a raw object ptr.
| ptr | The object pointer |
| BaseType | The reference type |
| ObjectType | The object type |
|
inline |
|
inline |
Get a reference type from a raw object ptr type.
It is always important to get a reference type if we want to return a value as reference or keep the object alive beyond the scope of the function.
| ptr | The object pointer |
| RefType | The reference type |
| ObjectType | The object type |
|
inline |
|
inline |
|
inline |
Allocate an object using default allocator.
| args | arguments to the constructor. |
| T | the node type. |
|
inline |
|
inline |
|
inline |
|
inline |
convert a string to TVM type.
| s | The string to be converted. |
|
inline |
|
inline |
|
inline |
Convert type code to its name.
| type_code | The type code . |
1.8.11