FEATURED · 精选文章

CANN/ge ES图构建器CreateInputs接口

发布时间 / 2026/9/10 22:19:46
来源 / 创域科博编辑部
栏目 / 资讯中心
CANN/ge ES图构建器CreateInputs接口 CreateInputs【免费下载链接】geGEGraph Engine是面向昇腾的图编译器和执行器提供了计算图优化、多流并行、内存复用和模型下沉等技术手段加速模型执行效率减少模型内存占用。 GE 提供对 PyTorch、TensorFlow 前端的友好接入能力并同时支持 onnx、pb 等主流模型格式的解析与编译。项目地址: https://gitcode.com/cann/ge产品支持情况全量芯片支持。头文件/库文件头文件#include ge/es_graph_builder.h库文件libeager_style_graph_builder_base.so、libeager_style_graph_builder_base_static.a功能说明创建指定数量的输入节点。函数原型创建指定数量的输入节点返回数组适用于编译时的结构化绑定template size_t inputs_num std::arrayEsTensorHolder, inputs_num CreateInputs(size_t start_index 0) - 创建指定数量的输入节点返回容器适用于运行时动态绑定 std::vectorEsTensorHolder CreateInputs(size_t inputs_num, size_t start_index 0) ## 参数说明 | 参数名 | 输入/输出 | 说明 | | --- | --- | --- | | inputs_num | 输入 | 输入节点数量。 | | start_index | 输入 | 输入节点起始索引如果不为0表示前面已经创建了其他输入节点整体输入节点的索引应该从0开始连续递增。 | ## 返回值说明 | 参数名 | 类型 | 说明 | | --- | --- | --- | | - | std::arrayEsTensorHolder, inputs_num | 返回创建的输入张量持有者数组。 | | - | std::vectorEsTensorHolder | 返回创建的输入张量持有者容器。 | ## 约束说明 无 ## 调用示例 - 创建指定数量的输入节点返回数组适用于编译时的结构化绑定 c // 1. 创建图构建器EsGraphBuilder EsGraphBuilder builder(graph_name); // 2. 添加 2 个输入节点使用c17的结构化绑定到data0和data1 EsTensorHolder [data0, data1] builder.CreateInputs2(); - 创建指定数量的输入节点返回容器适用于运行时动态绑定 // 1. 创建图构建器EsGraphBuilder EsGraphBuilder builder(graph_name); // 2. 添加 2 个输入节点 std::vectorEsTensorHolder inputs builder.CreateInputs(2); 【免费下载链接】geGEGraph Engine是面向昇腾的图编译器和执行器提供了计算图优化、多流并行、内存复用和模型下沉等技术手段加速模型执行效率减少模型内存占用。 GE 提供对 PyTorch、TensorFlow 前端的友好接入能力并同时支持 onnx、pb 等主流模型格式的解析与编译。项目地址: https://gitcode.com/cann/ge创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考
RELATED — 相关阅读

相关资讯

LATEST — 最新资讯

最新发布

TODAY — 本日精选

新闻

WEEKLY — 本周精选

新闻

MONTHLY — 本月精选

新闻