昇腾社区首页
中文
注册

函数:get_op_attr

产品支持情况

产品

是否支持

Atlas A3 训练系列产品/Atlas A3 推理系列产品

Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件

Atlas 训练系列产品

Atlas 推理系列产品

Atlas 200I/500 A2 推理产品

Atlas 200/300/500 推理产品

功能说明

获取整网中某个模型中某个算子的属性的值。

函数原型

  • C函数原型
    1
    const char *aclmdlGetOpAttr(aclmdlDesc *modelDesc, const char *opName, const char *attr)
    
  • python函数
    1
    attr_value = acl.mdl.get_op_attr(model_desc, op_name, attr)
    

参数说明

参数名

说明

model_desc

int,aclmdlDesc类型的指针地址。

需提前调用 acl.mdl.create_desc接口创建aclmdlDesc类型的数据,再调用 acl.mdl.get_desc接口根据模型ID获取到对应的aclmdlDesc类型的数据。

op_name

str,算子名称。

attr

str,算子属性。

当前仅支持“_datadump_original_op_names”属性,用于记录某个算子是由哪些算子融合得到的。通过本接口获取到的“_datadump_original_op_names”属性值格式为[opName1_len]opName1…..[opNameN_len]opNameN,“opNameN_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

返回值说明

返回值

说明

attr_value

str,返回属性值的字符串,若opName或者attr属性不存在或者attr属性值为空,均返回空字符串。

若调用该接口失败,则返回None。