文档管理中心
您当前浏览的HarmonyOS 5.0.0(API 12)文档归档不再维护,推荐您使用最新版本。详细请参考文档维护策略变更

hiai_tensor.h

本文导读

概述

模型推理时使用的输入输出内存相关的辅助接口。

通过以下接口,可以关联AippParam到tensor中,也可计算图片格式需要申请的tensor内存大小。

库: libhiai_foundation.so

系统能力: SystemCapability.AI.HiAIFoundation

起始版本: 4.1.0(11)

相关模块:HiAIFoundation

汇总

函数

展开

名称

描述

size_t HMS_HiAITensor_GetSizeWithImageFormat (NN_TensorDesc *desc, HiAI_ImageFormat format)

根据NN_TensorDesc和HiAI_ImageFormat计算申请tensor的大小。

OH_NN_ReturnCode HMS_HiAITensor_SetAippParams (NN_Tensor *tensor, HiAI_AippParam *aippParams[], size_t aippNum)

给NN_Tensor设置AippParams。

在 API参考 中进行搜索
请输入您想要搜索的关键词