BuildAndReset【免费下载链接】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功能说明构建计算图。函数原型构建计算图std::unique_ptrGraph BuildAndReset() const - 根据输出张量列表构建计算图 std::unique_ptrGraph BuildAndReset(const std::vectorEsTensorHolder outputs) ## 参数说明 | 参数名 | 输入/输出 | 说明 | | --- | --- | --- | | outputs | 输入 | 输出张量持有者向量。 | ## 返回值说明 | 参数名 | 类型 | 说明 | | --- | --- | --- | | - | std::unique_ptrGraph | 返回构建的图智能指针失败时返回nullptr。调用后图构建器将不再可用。 | ## 约束说明 无 ## 调用示例 - 构建计算图 c // 1. 创建图构建器EsGraphBuilder EsGraphBuilder builder(graph_name); // 2. 添加 2 个输入节点 EsTensorHolder [data0, data1] builder.CreateInputs2(); // 3. 添加中间节点C中加减乘除等常用运算符被重载可以直接使用 EsTensorHolder add data0 data1; // 4. 设置图输出 builder.SetOutput(add, 0); // 5. 完成构图获取构造好的Graph对象builder中的资源随析构而销毁 std::unique_ptrge::Graph graph builder.BuildAndReset(); - 根据输出张量列表构建计算图 // 1. 创建图构建器EsGraphBuilder EsGraphBuilder builder(graph_name); // 2. 添加 2 个输入节点 EsTensorHolder [data0, data1] builder.CreateInputs2(); // 3. 添加中间节点C中加减乘除等常用运算符被重载可以直接使用 // 4. 设置图输出 // 5. 完成构图获取构造好的Graph对象builder中的资源随析构而销毁 std::unique_ptrge::Graph graph builder.BuildAndReset({data0 data1}); // 一行中完成345 【免费下载链接】geGEGraph Engine是面向昇腾的图编译器和执行器提供了计算图优化、多流并行、内存复用和模型下沉等技术手段加速模型执行效率减少模型内存占用。 GE 提供对 PyTorch、TensorFlow 前端的友好接入能力并同时支持 onnx、pb 等主流模型格式的解析与编译。项目地址: https://gitcode.com/cann/ge创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考