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; };

参数名

输入/输出

描述

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：成功。其他值：失败。

参数名

类型

描述

graphStatus

GRAPH_SUCCESS：成功。

其他值：失败。

调用示例

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

父主题： Graph编译接口