ARTICLE DETAIL

资讯详情

深耕网站视觉设计与运营推广的一线实战洞察。

昇腾CANN/GE算子属性获取API

昇腾CANN/GE算子属性获取API aclmdlGetOpAttr【免费下载链接】geGEGraph Engine是面向昇腾的图编译器和执行器提供了计算图优化、多流并行、内存复用和模型下沉等技术手段加速模型执行效率减少模型内存占用。 GE 提供对 PyTorch、TensorFlow 前端的友好接入能力并同时支持 onnx、pb 等主流模型格式的解析与编译。项目地址: https://gitcode.com/cann/ge产品支持情况Ascend 950PR/Ascend 950DT支持Atlas A3 训练系列产品/Atlas A3 推理系列产品支持Atlas A2 训练系列产品/Atlas A2 推理系列产品支持Atlas 200I/500 A2 推理产品支持Atlas 推理系列产品支持Atlas 训练系列产品支持IPV350不支持功能说明获取整网中某个模型中某个算子的属性的值。函数原型const char *aclmdlGetOpAttr(aclmdlDesc *modelDesc, const char *opName, const char *attr)参数说明参数名输入/输出说明modelDesc输入aclmdlDesc类型的指针。需提前调用aclmdlCreateDesc接口创建aclmdlDesc类型的数据再调用aclmdlGetDesc接口根据模型ID获取到对应的aclmdlDesc类型的数据。opName输入算子名称的指针。attr输入算子属性的指针。当前仅支持_datadump_original_op_names属性用于记录某个算子是由哪些算子融合得到的。通过本接口获取到的_datadump_original_op_names属性值格式为[opName1_len]opName1…..[opNameN_len]opNameNopNameN_len表示算子名称字符串的长度。_datadump_original_op_names属性值示例如下表示某个融合算子由scale2c_branch2c、bn2c_branch2c、res2c_branch2c、res2c、res2c_relu这五个算子融合而成的算子名称字符串的长度分别为16、13、14、5、10[16]scale2c_branch2c[13]bn2c_branch2c[14]res2c_branch2c[5]res2c[10]res2c_relu返回值说明返回属性值的字符串指针该指针的生命周期与modelDesc相同若modelDesc资源被销毁则该指针指向的内容也会自动被销毁。若opName或者attr属性不存在、或者attr属性值为空均返回指向空字符串的指针。若调用该接口失败则返回nullptr。【免费下载链接】geGEGraph Engine是面向昇腾的图编译器和执行器提供了计算图优化、多流并行、内存复用和模型下沉等技术手段加速模型执行效率减少模型内存占用。 GE 提供对 PyTorch、TensorFlow 前端的友好接入能力并同时支持 onnx、pb 等主流模型格式的解析与编译。项目地址: https://gitcode.com/cann/ge创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考
返回列表