CANN/ge GraphProcessPoint API文档

GraphProcessPoint【免费下载链接】geGEGraph Engine是面向昇腾的图编译器和执行器提供了计算图优化、多流并行、内存复用和模型下沉等技术手段加速模型执行效率减少模型内存占用。 GE 提供对 PyTorch、TensorFlow 前端的友好接入能力并同时支持 onnx、pb 等主流模型格式的解析与编译。项目地址: https://gitcode.com/cann/ge产品支持情况产品是否支持Atlas A3 训练系列产品/Atlas A3 推理系列产品√Atlas A2 训练系列产品/Atlas A2 推理系列产品√函数功能GraphProcessPoint构造函数返回一个GraphProcessPoint对象。函数原型GraphProcessPoint(framework, graph_file, load_params{}, compile_config_path, nameNone)参数说明参数名称数据类型取值说明frameworkFrameworkIR文件的框架类型详见dataflow.Framework。graph_filestrIR文件路径。load_paramsDict[str, str]配置参数map映射表key为参数类型value为参数值均为String格式用于描述原始模型解析参数。compile_config_pathstr编译graph时的配置文件路径。配置文件实例如下{build_option:{},inputs_tensor_desc:[{data_type:DT_UINT32,shape:[3]},{data_type:DT_UINT32,shape:[3]}]}namestr处理点名称框架会自动保证名称唯一不设置时会自动生成GraphProcessPoint, GraphProcessPoint_1, GraphProcessPoint_2,...的名称。表 1GraphProcessPoint的json配置文件配置项可选/必选描述build_options可选值为mapstring, string, 有需要设置时参考Ascend Graph中的build_options。inputs_tensor_desc可选值为listGraph的输入节点list元素为tensor的描述。inputs_tensor_desc.data_type可选字符串类型。取值为Graph中的data_type对应字符串。inputs_tensor_desc.shape可选值为整数类型的列表。取值为Graph中的shape。返回值正常场景下返回None。异常情况下会抛出DfException异常。可以通过捕捉异常获取DfException中的error_code与message查看具体的错误码及错误信息。详细信息请参考DataFlow错误码。调用示例import dataflow as df pp1 df.GraphProcessPoint(...)约束说明无【免费下载链接】geGEGraph Engine是面向昇腾的图编译器和执行器提供了计算图优化、多流并行、内存复用和模型下沉等技术手段加速模型执行效率减少模型内存占用。 GE 提供对 PyTorch、TensorFlow 前端的友好接入能力并同时支持 onnx、pb 等主流模型格式的解析与编译。项目地址: https://gitcode.com/cann/ge创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考