harmony 鸿蒙ModelTypes.idl
ModelTypes.idl
概述
该文件定义AI模型相关的结构体。
在PrepareModel阶段，需要解析Model并将其转换为用于推理的模型结构，在Run阶段则会执行模型推理。大致流程如下：
编写NodeAttrTypes.idl文件中每一个算子的函数，并将函数与NodeType进行关联。
遍历Model的subGraph参数，然后从子图的nodeIndecies中获得该子图包含的算子节点以及算子的输入输出张量和整个Model的输入输出张量。
通过Node的nodeType参数找到算子函数，并构建用于运行时的模型结构。
执行模型推理时，通过用户输入张量传递给模型并执行模型推理，最终输出模型推理的结果。
Since:
3.2
Version:
1.0
相关模块:
汇总
类
| 名称 | 描述 |
|---|---|
| Tensor | struct 张量结构体。 |
| Node | struct 算子节点结构体。 |
| SubGraph | struct 子图结构体。 |
| Model | struct 模型结构体。 |
关键字
| 名称 | 描述 |
|---|---|
| package ohos.hdi.nnrt.v2_0 | NNRt模块的包路径。 |
你可能感兴趣的鸿蒙文章
harmony 鸿蒙ActivityRecognitionTypes.idl
0
赞
- 所属分类： 后端技术
- 本文标签：
热门推荐
-
2、 - 优质文章
-
3、 gate.io
-
8、 golang
-
9、 openharmony
-
10、 Vue中input框自动聚焦