NVIDIA Media Interface: Tensor Processing
Description: This file contains the NvMedia Tensor Processing API.
Definition in file nvmedia_tensor.h.
Go to the source code of this file.
Data Structures | |
| struct | NvMediaTensorTaskStatus |
| Holds the status of the latest operation for a tensor. More... | |
| struct | NvMediaTensorAttr |
| Holds tensor creation attributes. More... | |
| struct | NvMediaTensorSurfaceMap |
| Defines the tensor surface map descriptor used by NvMediaTensorLock(). More... | |
Macros | |
| #define | NVMEDIA_TENSOR_VERSION_MAJOR (1u) |
| Major version number. More... | |
| #define | NVMEDIA_TENSOR_VERSION_MINOR (13u) |
| Minor version number. More... | |
| #define | NVMEDIA_TENSOR_VERSION_PATCH (0u) |
| Patch version number. More... | |
| #define | NVMTENSOR_4D_MAX_N (65536u) |
| Defines the maximum supported number of tensor surfaces. More... | |
| #define | NVMTENSOR_4D_MAX_C (8192u) |
| Defines the maximum supported number of channels of a 4D tensor. More... | |
| #define | NVMTENSOR_4D_MAX_H (8192u) |
| Defines the maximum supported height of a 4D tensor. More... | |
| #define | NVMTENSOR_4D_MAX_W (8192u) |
| Defines the maximum supported width of a 4D tensor. More... | |
| #define | NVMTENSOR_4D_MAX_X (1024u) |
| Defines the maximum supported x value of a 4D tensor. More... | |
| #define | NVMEDIA_TENSOR_TIMEOUT_INFINITE |
| This macro tells NvMediaTensorGetStatus() and NvMediaTensorLock() APIs to block infinitely till the operation finishes. More... | |
| #define | NVM_TENSOR_ATTR_CPU_ACCESS_UNCACHED (0x00000001u) |
| Specifies that the tensor CPU accesses are uncached. More... | |
| #define | NVM_TENSOR_ATTR_CPU_ACCESS_CACHED (0x00000002u) |
| Specifies that the tensor CPU accesses are cacheable. More... | |
| #define | NVM_TENSOR_ATTR_CPU_ACCESS_UNMAPPED (0x00000003u) |
| Specifies that the tensor CPU accesses are unmapped from the virtual address space of the current process. More... | |
| #define | NVM_TENSOR_ATTR_ALLOC_NONE (0x00000000u) |
| Specifies that the tensor allocation is on Soc DRAM. More... | |
| #define | NVM_TENSOR_INIT_ATTR(x) |
| A helper macro to initialize tensor creation attributes. More... | |
| #define | NVM_TENSOR_DEFINE_ATTR(x) |
| A helper macro to define tensor creation attributes. More... | |
| #define | NVM_TENSOR_SET_ATTR_4D(attr, N, C, H, W, order, datatype, bpe, accesstype, alloctype, X) |
| A helper macro to set 4-D tensor creation attributes. More... | |
Typedefs | |
| typedef struct NvMediaTensor | NvMediaTensor |
| A handle representing tensor objects. More... | |
Enumerations | |
| enum | NvMediaTensorAttrType { NVM_TENSOR_ATTR_DATA_TYPE = 0, NVM_TENSOR_ATTR_BITS_PER_ELEMENT, NVM_TENSOR_ATTR_DIMENSION_ORDER, NVM_TENSOR_ATTR_CPU_ACCESS, NVM_TENSOR_ATTR_ALLOC_TYPE, NVM_TENSOR_ATTR_4D_N, NVM_TENSOR_ATTR_4D_C, NVM_TENSOR_ATTR_4D_H, NVM_TENSOR_ATTR_4D_W, NVM_TENSOR_ATTR_4D_X, NVM_TENSOR_ATTR_MAX } |
| Defines attribute types for creating NvMedia Tensor. More... | |
| enum | NvMediaTensorLockAccess { NVMEDIA_TENSOR_ACCESS_READ = (1 << 0), NVMEDIA_TENSOR_ACCESS_WRITE = (1 << 1), NVMEDIA_TENSOR_ACCESS_READ_WRITE = (NVMEDIA_TENSOR_ACCESS_READ | NVMEDIA_TENSOR_ACCESS_WRITE) } |
| Defines tensor lock access types. More... | |
Functions | |
| void | NvMediaTensorDestroy (NvMediaTensor *tensor) |
| Destroys a tensor object previously created by NvMediaTensorCreateFromNvSciBuf(). More... | |
| NvMediaStatus | NvMediaTensorLock (NvMediaTensor *tensor, NvMediaTensorLockAccess lockAccessType, NvMediaTensorSurfaceMap *surfaceMap) |
| Locks a tensor and returns the associated mapped pointers pointing to the tensor surface data. More... | |
| void | NvMediaTensorUnlock (NvMediaTensor *tensor) |
| Unlocks a tensor. More... | |
| NvMediaStatus | NvMediaTensorGetStatus (NvMediaTensor *tensor, uint32_t millisecondWait, NvMediaTensorTaskStatus *status) |
| Gets the status of the last operation for the tensor, and optionally waits for the operation to complete or time out. More... | |
| NvMediaStatus | NvMediaTensorGetMetaData (const NvMediaTensor *tensor, NvMediaTensorMetaData *tensormetadata) |
| Fills in the metadata information for the tensor. More... | |
| NvMediaStatus | NvMediaTensorGetVersion (NvMediaVersion *version) |
| Returns version information for the NvMediaTensor library. More... | |
