harmony 鸿蒙model.h
model.h
概述
提供了模型相关接口，可以用于模型创建、模型推理等。
起始版本：
9
相关模块：
汇总
结构体
| 结构体名称 | 描述 |
|---|---|
| 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 | 回调函数指针。 |
函数
| 函数名称 | 描述 |
|---|---|
| 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) | 通过张量名获取模型的输出张量。 |
你可能感兴趣的鸿蒙文章
0
赞
- 所属分类： 后端技术
- 本文标签：
热门推荐
-
2、 - 优质文章
-
3、 gate.io
-
6、 golang
-
7、 openharmony