BuildAndReset【免费下载链接】geGEGraph Engine是面向昇腾的图编译器和执行器提供了计算图优化、多流并行、内存复用和模型下沉等技术手段加速模型执行效率减少模型内存占用。 GE 提供对 PyTorch、TensorFlow 前端的友好接入能力并同时支持 onnx、pb 等主流模型格式的解析与编译。项目地址: https://gitcode.com/cann/ge产品支持情况产品是否支持Atlas A3 训练系列产品 / Atlas A3 推理系列产品√Atlas A2 训练系列产品 / Atlas A2 推理系列产品√功能说明构建计算图。函数原型构建计算图std::unique_ptrGraph BuildAndReset() const根据输出张量列表构建计算图std::unique_ptrGraph BuildAndReset(const std::vectorEsTensorHolder outputs)参数说明参数名输入/输出说明outputs输入输出张量持有者向量。返回值说明参数名类型说明-std::unique_ptrGraph返回构建的图智能指针失败时返回nullptr。调用后图构建器将不再可用。约束说明无调用示例构建计算图// 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），仅供参考