model.h

概述

提供了模型相关接口,可以用于模型创建、模型推理等。

起始版本:

9

相关模块:

MindSpore

汇总

结构体

结构体名称 描述
OH_AI_TensorHandleArray 张量数组结构体,用于存储张量数组指针和张量数组长度
OH_AI_ShapeInfo 维度信息,最大的维度为MS_MAX_SHAPE_NUM
OH_AI_CallBackParam 回调函数中传入的算子信息。

宏定义

宏定义名称 描述
OH_AI_MAX_SHAPE_NUM 32 维度信息,最大的维度为MS_MAX_SHAPE_NUM

类型定义

类型定义名称 描述
OH_AI_ModelHandle 指向模型对象的指针。
OH_AI_TensorHandleArray 张量数组结构体,用于存储张量数组指针和张量数组长度
OH_AI_ShapeInfo 维度信息,最大的维度为MS_MAX_SHAPE_NUM
OH_AI_CallBackParam 回调函数中传入的算子信息。
OH_AI_KernelCallBack) (const OH_AI_TensorHandleArray inputs, const OH_AI_TensorHandleArray outputs, const OH_AI_CallBackParam kernel_Info) 回调函数指针。

函数

函数名称 描述
OH_AI_ModelCreate () 创建一个模型对象。
OH_AI_ModelDestroy (OH_AI_ModelHandle *model) 释放一个模型对象。
OH_AI_ModelBuild (OH_AI_ModelHandle model, const void *model_data,
size_t data_size, OH_AI_ModelType model_type,
const OH_AI_ContextHandle model_context)
从内存缓冲区加载并编译MindSpore模型。注意,同一个OH_AI_ContextHandle对象仅能传递给OH_AI_ModelBuildFromFile或者OH_AI_ModelBuild一次,如果多次调用该函数需要创建多个不同的OH_AI_ContextHandle
OH_AI_ModelBuildFromFile (OH_AI_ModelHandle model,
const char *model_path, OH_AI_ModelType model_type, const OH_AI_ContextHandle model_context)
通过模型文件加载并编译MindSpore模型。注意,同一个OH_AI_ContextHandle对象仅能传递给OH_AI_ModelBuildFromFile或者OH_AI_ModelBuild一次,如果多次调用该函数需要创建多个不同的OH_AI_ContextHandle
OH_AI_ModelResize (OH_AI_ModelHandle model, const OH_AI_TensorHandleArray inputs,
OH_AI_ShapeInfo *shape_infos, size_t shape_info_num)
调整已编译模型的输入形状。
OH_AI_ModelPredict (OH_AI_ModelHandle model, const OH_AI_TensorHandleArray inputs, OH_AI_TensorHandleArray *outputs, const OH_AI_KernelCallBack before, const OH_AI_KernelCallBack after) 执行模型推理。
OH_AI_ModelGetInputs (const OH_AI_ModelHandle model) 获取模型的输入张量数组结构体。
OH_AI_ModelGetOutputs (const OH_AI_ModelHandle model) 获取模型的输出张量数组结构体。
OH_AI_ModelGetInputByTensorName (const OH_AI_ModelHandle model, const char *tensor_name) 通过张量名获取模型的输入张量。
OH_AI_ModelGetOutputByTensorName (const OH_AI_ModelHandle model, const char *tensor_name) 通过张量名获取模型的输出张量。