Eager Style Graph Builder 类关系文档【免费下载链接】geGEGraph Engine是面向昇腾的图编译器和执行器提供了计算图优化、多流并行、内存复用和模型下沉等技术手段加速模型执行效率减少模型内存占用。 GE 提供对 PyTorch、TensorFlow 前端的友好接入能力并同时支持 onnx、pb 等主流模型格式的解析与编译。项目地址: https://gitcode.com/cann/ge概述Eager Style Graph Builder 是 GraphEngine 中用于构建计算图的函数式接口模块提供了便捷的图构建功能。该模块对外头文件位于inc/external/ge/eager_style_graph_builder/目录下。目录结构inc/external/ge/eager_style_graph_builder/ ├── c/ │ └── esb_funcs.h # C语言接口函数声明 └── cpp/ ├── compliant_node_builder.h # 合规节点构建器类定义 ├── es_graph_builder.h # 图构建器类定义 ├── es_tensor_holder.h # 张量持有者类定义 ├── es_tensor_like.h # 类Tensor类型定义 ├── es_c_graph_builder.h # C风格图构建器类定义 └── es_c_tensor_holder.h # C风格张量持有者类定义核心类关系图主要类详细说明1. EsGraphBuilder 类文件位置:cpp/es_graph_builder.h功能: 图构建器类用于构建和管理计算图主要方法:CreateInput()- 创建图输入节点CreateInputs()- 批量创建图默认格式的输入节点CreateTensor()- 按运行时DataType创建指定形状的张量CreateVector()- 创建向量常量CreateScalar()- 创建标量常量CreateVariable()- 创建变量SetAttr()- 设置图属性SetOutput()- 设置图输出BuildAndReset()- 构建计算图关系:封装了EsCGraphBuilderC结构体创建和管理EsTensorHolder对象最终构建出ge::Graph对象2. EsTensorHolder 类文件位置:cpp/es_tensor_holder.h功能: 张量持有者类封装了张量的各种操作主要方法:算术运算operator,operator-,operator*,operator/属性设置SetDataType(),SetFormat(),SetShape()属性管理SetAttr(),SetAttrForNode()访问器GetCTensorHolder(),GetProducer()关系:封装了EsCTensorHolderC结构体与ge::GNode关联通过GetProducer()支持链式调用模式3. EsTensorLike 类文件位置:cpp/es_tensor_like.h功能类 Tensor 类型定义用于将 EsTensorHolder、标量和向量转为 EsTensorHolder 对象主要方法:ToTensorHolder()- 转为EsTensorHolder对象GetOwnerBuilder()- 获取对应Tensor的owner builder4. CompliantNodeBuilder 类文件位置:cpp/compoliant_node_builder.h功能: 合规节点构建器类用于构建符合IR规范的图节点主要方法:OpType()- 设置操作符类型IrDefInputsV2()- 定义ABI安全的IR输入规范IrDefOutputsV2()- 定义ABI安全的IR输出规范IrDefAttrsV2()- 定义ABI安全的IR属性规范Name()- 设置节点名称InstanceDynamicInputNum()- 设置动态输入实例数量InstanceDynamicOutputNum()- 设置动态输出实例数量InstanceOutputDataType()- 设置输出数据类型InstanceOutputShape()- 设置输出形状InstanceOutputOriginShape()- 设置输出原始形状InstanceOutputStorageShape()- 设置输出存储形状InstanceOutputFormat()- 设置输出格式InstanceOutputOriginFormat()- 设置输出原始格式InstanceOutputStorageFormat()- 设置输出存储格式Build()- 构建并返回图节点5. C API 函数文件位置:c/esb_funcs.h功能提供底层的 C 语言接口主要函数类别:图构建器管理EsCreateGraphBuilder(),EsDestroyGraphBuilder()输入创建EsCreateGraphInput(),EsCreateGraphInputWithDetails()常量创建EsCreateScalar*(),EsCreateVector*(),EsCreateConst*()属性设置EsSet*AttrForGraph(),EsSet*AttrForTensor(),EsSet*AttrForNode()输出设置EsSetGraphOutput()图构建EsBuildGraphAndReset()使用示例参考sample【免费下载链接】geGEGraph Engine是面向昇腾的图编译器和执行器提供了计算图优化、多流并行、内存复用和模型下沉等技术手段加速模型执行效率减少模型内存占用。 GE 提供对 PyTorch、TensorFlow 前端的友好接入能力并同时支持 onnx、pb 等主流模型格式的解析与编译。项目地址: https://gitcode.com/cann/ge创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考