|
NVIDIA DRIVE OS Linux SDK API Reference
|
5.1.15.2 Release For Test and Development only
|
Go to the documentation of this file.
16 #ifndef NVMEDIA_TENSOR_H
17 #define NVMEDIA_TENSOR_H
45 #define NVMEDIA_TENSOR_VERSION_MAJOR (1u)
47 #define NVMEDIA_TENSOR_VERSION_MINOR (7u)
53 #define NVMEDIA_TENSOR_TIMEOUT_INFINITE (0xFFFFFFFFu)
103 #define NVM_TENSOR_ATTR_CPU_ACCESS_UNCACHED (0x00000001u)
107 #define NVM_TENSOR_ATTR_CPU_ACCESS_CACHED (0x00000002u)
112 #define NVM_TENSOR_ATTR_CPU_ACCESS_UNMAPPED (0x00000003u)
117 #define NVM_TENSOR_ATTR_ALLOC_RESERVED (0x00000010u)
121 #define NVM_TENSOR_ATTR_ALLOC_NONE (0x00000000u)
138 #define NVM_TENSOR_INIT_ATTR(x) \
140 x[0].type = NVM_TENSOR_ATTR_DATA_TYPE; \
143 x[1].type = NVM_TENSOR_ATTR_4D_N; \
146 x[2].type = NVM_TENSOR_ATTR_4D_C; \
149 x[3].type = NVM_TENSOR_ATTR_4D_H; \
152 x[4].type = NVM_TENSOR_ATTR_4D_W; \
155 x[5].type = NVM_TENSOR_ATTR_4D_X; \
158 x[6].type = NVM_TENSOR_ATTR_BITS_PER_ELEMENT; \
161 x[7].type = NVM_TENSOR_ATTR_DIMENSION_ORDER; \
164 x[8].type = NVM_TENSOR_ATTR_CPU_ACCESS; \
167 x[9].type = NVM_TENSOR_ATTR_ALLOC_TYPE; \
174 #define NVM_TENSOR_DEFINE_ATTR(x) \
175 NvMediaTensorAttr x[NVM_TENSOR_ATTR_MAX]; \
176 NVM_TENSOR_INIT_ATTR(x); \
198 #define NVM_TENSOR_SET_ATTR_4D(attr, N, C, H, W, order, datatype, bpe, accesstype, alloctype, X)\
200 attr[0].type = NVM_TENSOR_ATTR_DATA_TYPE; \
201 attr[0].value = NVM_TENSOR_ATTR_DATA_TYPE_##datatype; \
203 attr[1].type = NVM_TENSOR_ATTR_4D_N; \
206 attr[2].type = NVM_TENSOR_ATTR_4D_C; \
209 attr[3].type = NVM_TENSOR_ATTR_4D_H; \
212 attr[4].type = NVM_TENSOR_ATTR_4D_W; \
215 attr[5].type = NVM_TENSOR_ATTR_4D_X; \
218 attr[6].type = NVM_TENSOR_ATTR_BITS_PER_ELEMENT; \
219 attr[6].value = NVM_TENSOR_ATTR_BITS_PER_ELEMENT_##bpe; \
221 attr[7].type = NVM_TENSOR_ATTR_DIMENSION_ORDER; \
222 attr[7].value = NVM_TENSOR_ATTR_DIMENSION_ORDER_##order; \
224 attr[8].type = NVM_TENSOR_ATTR_CPU_ACCESS; \
225 attr[8].value = NVM_TENSOR_ATTR_CPU_ACCESS_##accesstype; \
227 attr[9].type = NVM_TENSOR_ATTR_ALLOC_TYPE; \
228 attr[9].value = NVM_TENSOR_ATTR_ALLOC_##alloctype; \
333 uint32_t millisecondWait,
@ NVMEDIA_TENSOR_ACCESS_WRITE
Write access.
NvMediaTensorLockAccess
Defines tensor lock access types.
NvMediaStatus NvMediaTensorLock(NvMediaTensor *tensor, NvMediaTensorLockAccess lockAccessType, NvMediaTensorSurfaceMap *surfaceMap)
Locks a tensor and returns the associated mapped pointers pointing to the tensor surface data.
@ NVMEDIA_TENSOR_ACCESS_READ
Read access.
NvMediaStatus
Defines all possible error codes.
@ NVMEDIA_TENSOR_ACCESS_READ_WRITE
Read/Write access.
void NvMediaTensorUnlock(NvMediaTensor *tensor)
Unlocks a tensor.