CANN/GE获取模型输出名称
aclmdlGetOutputNameByIndex【免费下载链接】geGEGraph Engine是面向昇腾的图编译器和执行器提供了计算图优化、多流并行、内存复用和模型下沉等技术手段加速模型执行效率减少模型内存占用。 GE 提供对 PyTorch、TensorFlow 前端的友好接入能力并同时支持 onnx、pb 等主流模型格式的解析与编译。项目地址: https://gitcode.com/cann/ge产品支持情况产品是否支持Atlas A3 训练系列产品 / Atlas A3 推理系列产品√Atlas A2 训练系列产品 / Atlas A2 推理系列产品√功能说明根据模型描述信息获取模型中指定输出的输出算子名称、算子输出边的下标、top名称或输出名称。函数原型const char *aclmdlGetOutputNameByIndex(const aclmdlDesc *modelDesc, size_t index)参数说明参数名输入/输出说明modelDesc输入aclmdlDesc类型的指针。需提前调用aclmdlCreateDesc接口创建aclmdlDesc类型的数据。index输入指定获取第几个输出的输出算子名称、算子输出边下标、top名称或输出名称index值从0开始。返回值说明返回指定输出的输出算子名称、算子输出边的下标、top名称或输出名称。不同原始网络、不同构建模型的方式调用本接口获取的返回值格式不同。Caffe网络返回值格式如下各项之间以冒号分割如果模型中包含top名称就返回不包含就不返回输出算子名称 : 算子输出边下标 : top名称TensorFlow网络使用ATC工具构建om模型的场景下返回值格式如下各项之间以冒号分割输出算子名称 : 算子输出边下标使用构图接口构建om模型的场景下返回值格式如下各项之间以下划线分割output_网络输出下标_输出算子名称_算子输出边下标构图接口的详细说明请参见《图模式开发指南》。ONNX网络在构建模型时不指定输出节点名称node_name或输出名称output的name或者仅指定输出名称返回值格式如下各项之间以冒号分割输出算子名称 : 算子输出边下标 : 输出名称在构建模型时指定输出节点名称node_name输出算子名称可能是图融合后的算子名称也可能是子图名称。使用ATC工具构建om模型的场景下返回值格式如下各项之间以冒号分割输出算子名称 : 算子输出边下标使用构图接口构建om模型的场景下返回值格式如下各项之间以下划线分割output_网络输出下标_输出算子名称_算子输出边下标构图接口的详细说明请参见《图模式开发指南》。同时指定输出节点名称node_name和输出名称output的name接口返回报错。【免费下载链接】geGEGraph Engine是面向昇腾的图编译器和执行器提供了计算图优化、多流并行、内存复用和模型下沉等技术手段加速模型执行效率减少模型内存占用。 GE 提供对 PyTorch、TensorFlow 前端的友好接入能力并同时支持 onnx、pb 等主流模型格式的解析与编译。项目地址: https://gitcode.com/cann/ge创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.coloradmin.cn/o/2602345.html
如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈,一经查实,立即删除!