aclgrphConvertToWeightRefreshableGraphs
产品支持情况
全量芯片支持。
头文件/库文件
- 头文件:#include <ge/ge_ir_build.h>
- 库文件:libge_compiler.so
函数原型
1
|
graphStatus aclgrphConvertToWeightRefreshableGraphs(const ge::Graph &origin_graph, const std::vector<AscendString> &const_names, WeightRefreshableGraphs &weight_refreshable_graphs) |
参数说明
|
参数名 |
输入/输出 |
说明 |
|---|---|---|
|
origin_graph |
输入 |
需要权重更新的原图。 |
|
const_names |
输入 |
需要权重更新的节点名数组。 |
|
weight_refreshable_graphs |
输出 |
权重可更新的图,该参数为一结构体,包括三部分:权重初始化图,权重更新图,推理图。内容详见如下结构体示例。 |
1 2 3 4 5 |
struct WeightRefreshableGraphs { ge::Graph infer_graph; ge::Graph var_init_graph; ge::Graph var_update_graph; }; |
调用示例
完整调用示例请参见调用示例。
父主题: aclgrph接口