Lines Matching refs:tensorDesc

65 | OH_NN_ReturnCode OH_NNModel_AddTensorToModel(OH_NNModel *model, const NN_TensorDesc *tensorDesc) …
97 | OH_NN_ReturnCode OH_NNTensorDesc_SetName(NN_TensorDesc *tensorDesc, const char *name) | 设置张量描述的名称…
98 | OH_NN_ReturnCode OH_NNTensorDesc_GetName(const NN_TensorDesc *tensorDesc, const char **name) | 获取…
99 | OH_NN_ReturnCode OH_NNTensorDesc_SetDataType(NN_TensorDesc *tensorDesc, OH_NN_DataType dataType) …
100 | OH_NN_ReturnCode OH_NNTensorDesc_GetDataType(const NN_TensorDesc *tensorDesc, OH_NN_DataType *dat…
101 | OH_NN_ReturnCode OH_NNTensorDesc_SetShape(NN_TensorDesc *tensorDesc, const int32_t *shape, size_t…
102 | OH_NN_ReturnCode OH_NNTensorDesc_GetShape(const NN_TensorDesc *tensorDesc, int32_t **shape, size_…
103 | OH_NN_ReturnCode OH_NNTensorDesc_SetFormat(NN_TensorDesc *tensorDesc, OH_NN_Format format) | 设置张量…
104 | OH_NN_ReturnCode OH_NNTensorDesc_GetFormat(const NN_TensorDesc *tensorDesc, OH_NN_Format *format)…
105 | OH_NN_ReturnCode OH_NNTensorDesc_GetElementCount(const NN_TensorDesc *tensorDesc, size_t *element…
106 | OH_NN_ReturnCode OH_NNTensorDesc_GetByteSize(const NN_TensorDesc *tensorDesc, size_t *byteSize) |…
107 | OH_NN_ReturnCode OH_NNTensorDesc_Destroy(NN_TensorDesc **tensorDesc) | 销毁张量描述实例。 |
113 | NN_Tensor* OH_NNTensor_Create(size_t deviceID, NN_TensorDesc *tensorDesc) | 从张量描述创建张量实例，会申请设备共享内存…
114 | NN_Tensor* OH_NNTensor_CreateWithSize(size_t deviceID, NN_TensorDesc *tensorDesc, size_t size) | …
115 | NN_Tensor* OH_NNTensor_CreateWithFd(size_t deviceID, NN_TensorDesc *tensorDesc, int fd, size_t si…
292         NN_TensorDesc* tensorDesc = OH_NNTensorDesc_Create();
293         CHECKEQ(tensorDesc, nullptr, -1, "Create TensorDesc failed.");
296         returnCode = OH_NNTensorDesc_SetShape(tensorDesc, inputDims, 4);
299         returnCode = OH_NNTensorDesc_SetDataType(tensorDesc, OH_NN_FLOAT32);
302         returnCode = OH_NNTensorDesc_SetFormat(tensorDesc, OH_NN_FORMAT_NONE);
305         returnCode = OH_NNModel_AddTensorToModel(model, tensorDesc);
312         tensorDesc = OH_NNTensorDesc_Create();
313         CHECKEQ(tensorDesc, nullptr, -1, "Create TensorDesc failed.");
315         returnCode = OH_NNTensorDesc_SetShape(tensorDesc, inputDims, 4);
318         returnCode = OH_NNTensorDesc_SetDataType(tensorDesc, OH_NN_FLOAT32);
321         returnCode = OH_NNTensorDesc_SetFormat(tensorDesc, OH_NN_FORMAT_NONE);
324         returnCode = OH_NNModel_AddTensorToModel(model, tensorDesc);
331         tensorDesc = OH_NNTensorDesc_Create();
332         CHECKEQ(tensorDesc, nullptr, -1, "Create TensorDesc failed.");
335         returnCode = OH_NNTensorDesc_SetShape(tensorDesc, &activationDims, 1);
338         returnCode = OH_NNTensorDesc_SetDataType(tensorDesc, OH_NN_INT8);
341         returnCode = OH_NNTensorDesc_SetFormat(tensorDesc, OH_NN_FORMAT_NONE);
344         returnCode = OH_NNModel_AddTensorToModel(model, tensorDesc);
356         tensorDesc = OH_NNTensorDesc_Create();
357         CHECKEQ(tensorDesc, nullptr, -1, "Create TensorDesc failed.");
359         returnCode = OH_NNTensorDesc_SetShape(tensorDesc, inputDims, 4);
362         returnCode = OH_NNTensorDesc_SetDataType(tensorDesc, OH_NN_FLOAT32);
365         returnCode = OH_NNTensorDesc_SetFormat(tensorDesc, OH_NN_FORMAT_NONE);
368         returnCode = OH_NNModel_AddTensorToModel(model, tensorDesc);