CANN/GE获取模型输出名称

aclmdlGetOutputNameByIndex

【免费下载链接】geGE（Graph Engine）是面向昇腾的图编译器和执行器，提供了计算图优化、多流并行、内存复用和模型下沉等技术手段，加速模型执行效率，减少模型内存占用。 GE 提供对 PyTorch、TensorFlow 前端的友好接入能力，并同时支持 onnx、pb 等主流模型格式的解析与编译。项目地址: https://gitcode.com/cann/ge

产品支持情况

功能说明

根据模型描述信息获取模型中指定输出的输出算子名称、算子输出边的下标、top名称或输出名称。

函数原型

const char *aclmdlGetOutputNameByIndex(const aclmdlDesc *modelDesc, size_t index)

参数说明

返回值说明

返回指定输出的输出算子名称、算子输出边的下标、top名称或输出名称。不同原始网络、不同构建模型的方式，调用本接口获取的返回值格式不同。

• Caffe网络

  返回值格式如下，各项之间以冒号分割，如果模型中包含top名称就返回，不包含就不返回：

  输出算子名称 : 算子输出边下标 : top名称

• TensorFlow网络

  • 使用ATC工具构建om模型的场景下，返回值格式如下，各项之间以冒号分割：

    输出算子名称 : 算子输出边下标

  • 使用构图接口构建om模型的场景下，返回值格式如下，各项之间以下划线分割：

    output_网络输出下标_输出算子名称_算子输出边下标

    构图接口的详细说明请参见《图模式开发指南》。

• ONNX网络

  • 在构建模型时，不指定输出节点名称（node_name）或输出名称（output的name），或者仅指定输出名称，返回值格式如下，各项之间以冒号分割：

    输出算子名称 : 算子输出边下标 : 输出名称

  • 在构建模型时，指定输出节点名称（node_name）：

    输出算子名称可能是图融合后的算子名称，也可能是子图名称。

    • 使用ATC工具构建om模型的场景下，返回值格式如下，各项之间以冒号分割：

      输出算子名称 : 算子输出边下标

    • 使用构图接口构建om模型的场景下，返回值格式如下，各项之间以下划线分割：

      output_网络输出下标_输出算子名称_算子输出边下标

      构图接口的详细说明请参见《图模式开发指南》。

  • 同时指定输出节点名称（node_name）和输出名称（output的name），接口返回报错。

【免费下载链接】geGE（Graph Engine）是面向昇腾的图编译器和执行器，提供了计算图优化、多流并行、内存复用和模型下沉等技术手段，加速模型执行效率，减少模型内存占用。 GE 提供对 PyTorch、TensorFlow 前端的友好接入能力，并同时支持 onnx、pb 等主流模型格式的解析与编译。项目地址: https://gitcode.com/cann/ge