文档
注册
评分
提单
论坛
小AI

aclgrphSetOpAttr

函数功能

用于支持设置node属性方式的改图。

函数原型

graphStatus aclgrphSetOpAttr(Graph &graph, aclgrphAttrType attr_type, const char_t *cfg_path);

约束说明

无。

参数说明

参数名

输入/输出

描述

graph

输入/输出

作为输入时,代表要通过属性方式修改的图;

作为输出时表示已经修改后的图。

attr_type

输入

attr_type的枚举类型定义:

enum aclgrphAttrType { ATTR_TYPE_KEEP_DTYPE = 0, ATTR_TYPE_WEIGHT_COMPRESS };

ATTR_TYPE_KEEP_DTYPE:保持模型编译时个别算子的计算精度不变。

ATTR_TYPE_WEIGHT_COMPRESS :模型编译时对个别算子进行weight压缩。

cfg_path

输出

配置文件路径。配置文件举例:

计算精度不变的文件格式为:

Opname1
Opname2
…

weight压缩文件格式为:

Opname1;Opname2

返回值

参数名

类型

描述

-

graphStatus

0:成功。

其他值:失败。

搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词