17 #ifndef NVM_TENSORMETADATA_H
18 #define NVM_TENSORMETADATA_H
41 #define NVMEDIA_TENSOR_MAX_DIMENSIONS (8u)
45 #define NVM_TENSOR_ATTR_DIMENSION_ORDER_NHWC (0x00000001u)
47 #define NVM_TENSOR_ATTR_DIMENSION_ORDER_NCHW (0x00000002u)
49 #define NVM_TENSOR_ATTR_DIMENSION_ORDER_NCxHWx (0x00000003u)
52 #define NVM_TENSOR_ATTR_DATA_TYPE_UINT (0x00000001u)
54 #define NVM_TENSOR_ATTR_DATA_TYPE_INT (0x00000002u)
56 #define NVM_TENSOR_ATTR_DATA_TYPE_FLOAT (0x00000003u)
59 #define NVM_TENSOR_ATTR_BITS_PER_ELEMENT_64 (64U)
61 #define NVM_TENSOR_ATTR_BITS_PER_ELEMENT_32 (32U)
63 #define NVM_TENSOR_ATTR_BITS_PER_ELEMENT_16 (16U)
65 #define NVM_TENSOR_ATTR_BITS_PER_ELEMENT_8 (8U)
100 #define NVM_TENSOR_NHWC_E_STRIDE_INDEX 0U
109 #define NVM_TENSOR_NHWC_C_STRIDE_INDEX NVM_TENSOR_NHWC_E_STRIDE_INDEX
118 #define NVM_TENSOR_NHWC_W_STRIDE_INDEX 1U
127 #define NVM_TENSOR_NHWC_H_STRIDE_INDEX 2U
136 #define NVM_TENSOR_NHWC_N_STRIDE_INDEX 3U
160 #define NVM_TENSOR_NCHW_E_STRIDE_INDEX 0U
170 #define NVM_TENSOR_NCHW_W_STRIDE_INDEX NVM_TENSOR_NCHW_E_STRIDE_INDEX
179 #define NVM_TENSOR_NCHW_H_STRIDE_INDEX 1U
187 #define NVM_TENSOR_NCHW_C_STRIDE_INDEX 2U
196 #define NVM_TENSOR_NCHW_N_STRIDE_INDEX 3U
222 #define NVM_TENSOR_NCxHWx_E_STRIDE_INDEX 0U
232 #define NVM_TENSOR_NCxHWx_X_STRIDE_INDEX NVM_TENSOR_NCxHWx_E_STRIDE_INDEX
241 #define NVM_TENSOR_NCxHWx_W_STRIDE_INDEX 1U
250 #define NVM_TENSOR_NCxHWx_H_STRIDE_INDEX 2U
259 #define NVM_TENSOR_NCxHWx_Cx_STRIDE_INDEX 3U
268 #define NVM_TENSOR_NCxHWx_N_STRIDE_INDEX 4U
283 #define NVM_TENSOR_NHWC_C_DIMSZ_INDEX 0U
290 #define NVM_TENSOR_NHWC_W_DIMSZ_INDEX 1U
297 #define NVM_TENSOR_NHWC_H_DIMSZ_INDEX 2U
304 #define NVM_TENSOR_NHWC_N_DIMSZ_INDEX 3U
317 #define NVM_TENSOR_NCHW_W_DIMSZ_INDEX 0U
324 #define NVM_TENSOR_NCHW_H_DIMSZ_INDEX 1U
331 #define NVM_TENSOR_NCHW_C_DIMSZ_INDEX 2U
338 #define NVM_TENSOR_NCHW_N_DIMSZ_INDEX 3U
352 #define NVM_TENSOR_NCxHWx_x_DIMSZ_INDEX 0U
359 #define NVM_TENSOR_NCxHWx_W_DIMSZ_INDEX 1U
366 #define NVM_TENSOR_NCxHWx_H_DIMSZ_INDEX 2U
373 #define NVM_TENSOR_NCxHWx_Cx_DIMSZ_INDEX 3U
380 #define NVM_TENSOR_NCxHWx_N_DIMSZ_INDEX 4U