NamedIoNodeBuilder构造函数
【免费下载链接】geGE(Graph Engine)是面向昇腾的图编译器和执行器,提供了计算图优化、多流并行、内存复用和模型下沉等技术手段,加速模型执行效率,减少模型内存占用。 GE 提供对 PyTorch、TensorFlow 前端的友好接入能力,并同时支持 onnx、pb 等主流模型格式的解析与编译。项目地址: https://gitcode.com/cann/ge
产品支持情况
| 产品 | 是否支持 |
|---|---|
| Atlas A3 训练系列产品/Atlas A3 推理系列产品 | √ |
| Atlas A2 训练系列产品/Atlas A2 推理系列产品 | √ |
头文件/库文件
- 头文件:#include <graph/named_io_node_builder.h>
- 库文件:libgraph.so、libgraph_static.a
功能说明
NamedIoNodeBuilder构造函数,创建基于输入、输出和属性名称构造图节点的Builder对象。
NamedIoNodeBuilder面向直接按照节点输入/输出名称构图的场景。调用方可指定节点实例的输入名、输出名、属性名及相关描述信息;构建时根据节点类型匹配已通过REG_OP注册的算子IR定义,校验输入/输出实例名称及顺序是否与该定义兼容,最终创建对应的图节点。
函数原型
explicit NamedIoNodeBuilder(Graph &graph) ~NamedIoNodeBuilder()参数说明
| 参数名 | 输入/输出 | 描述 |
|---|---|---|
| graph | 输入 | 所属的图对象,构建完成的节点将被添加到此图中。 |
返回值说明
构造函数返回NamedIoNodeBuilder类型的对象。
约束说明
- NamedIoNodeBuilder对象不可拷贝构造和赋值。
- 构建器对象Build成功后不应再次使用。
【免费下载链接】geGE(Graph Engine)是面向昇腾的图编译器和执行器,提供了计算图优化、多流并行、内存复用和模型下沉等技术手段,加速模型执行效率,减少模型内存占用。 GE 提供对 PyTorch、TensorFlow 前端的友好接入能力,并同时支持 onnx、pb 等主流模型格式的解析与编译。项目地址: https://gitcode.com/cann/ge
创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考