Build【免费下载链接】geGEGraph Engine是面向昇腾的图编译器和执行器提供了计算图优化、多流并行、内存复用和模型下沉等技术手段加速模型执行效率减少模型内存占用。 GE 提供对 PyTorch、TensorFlow 前端的友好接入能力并同时支持 onnx、pb 等主流模型格式的解析与编译。项目地址: https://gitcode.com/cann/ge产品支持情况全量芯片支持。头文件/库文件头文件#include graph/named_io_node_builder.h库文件libgraph.so、libgraph_static.a功能说明构建图节点并添加到Graph。Build会基于已设置的输入、输出和属性信息创建GNode并校验输入/输出实例名称及顺序是否与已注册的算子IR定义兼容。成功时返回的GNode已被添加到Graph中。Build执行以下对外可见校验和操作校验Type已设置且节点类型对应的算子IR已通过REG_OP完成注册。校验输入/输出实例名称及顺序是否与已注册的算子IR定义兼容。创建GNode并添加到Graph。返回unique_ptrGNode成功时清空error_message。函数原型std::unique_ptrGNode Build(AscendString error_message)参数说明参数名输入/输出说明error_message输出构建失败时的错误信息。成功时error_message被清空。返回值说明参数名类型说明-std::unique_ptrGNode成功时返回GNode的unique_ptr已被添加到Graph中。失败时返回nullptr。约束说明Builder对象Build成功后不应再次使用Build成功后再次调用返回nullptr并输出错误信息Build() has already been called on this builder。节点类型对应的算子IR需已通过REG_OP完成注册否则Build返回nullptr。输入/输出实例需按IR定义顺序添加否则Build返回nullptr。Data和NetOutput节点不进行输入/输出实例兼容性校验。调用示例ge::Graph graph(test_graph); ge::AscendString error_msg; // 构建一个Add算子节点 auto node ge::NamedIoNodeBuilder(graph) .Type(Add) .Name(add_node) .AddInput(x1) .AddInput(x2) .AddOutput(y) .Build(error_msg); if (node ! nullptr) { // 构建成功节点已被添加到graph中 } else { // 构建失败查看error_msg获取错误信息 }【免费下载链接】geGEGraph Engine是面向昇腾的图编译器和执行器提供了计算图优化、多流并行、内存复用和模型下沉等技术手段加速模型执行效率减少模型内存占用。 GE 提供对 PyTorch、TensorFlow 前端的友好接入能力并同时支持 onnx、pb 等主流模型格式的解析与编译。项目地址: https://gitcode.com/cann/ge创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考