下载
中文
注册
我要评分
文档获取效率
文档正确性
内容完整性
文档易理解
在线提单
论坛求助
昇腾小AI

aclgrphConvertToWeightRefreshableGraphs

函数功能

该接口适用于权重更新场景,详细介绍请参见专题>权重更新

通过传入const节点名数组将原图转换成一组权重可更新的图。

函数原型

graphStatus aclgrphConvertToWeightRefreshableGraphs(const ge::Graph &origin_graph, const std::vector<AscendString> &const_names, WeightRefreshableGraphs &weight_refreshable_graphs)

参数说明

参数名

输入/输出

描述

origin_graph

输入

需要权重更新的原图。

const_names

输入

需要权重更新的节点名数组。

weight_refreshable_graphs

输出

权重可更新的图,该参数为一结构体,包括三部分:权重初始化图,权重更新图,推理图。

struct WeightRefreshableGraphs {
  ge::Graph infer_graph;
  ge::Graph var_init_graph;
  ge::Graph var_update_graph;
};

返回值

参数名

类型

描述

-

graphStatus

GRAPH_SUCCESS:成功。

其他值:失败。

调用示例

完整调用示例请参见调用示例

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

当前产品无相关内容

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