aclgrphConvertToWeightRefreshableGraphs

产品支持情况

产品	是否支持
Atlas A2 训练系列产品/Atlas 800I A2 推理产品	√
Atlas 200I/500 A2 推理产品	√
Atlas 推理系列产品	√
Atlas 训练系列产品	√
Atlas 200/300/500 推理产品	√

功能说明

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

参数名

类型

说明

graphStatus

GRAPH_SUCCESS：成功。

其他值：失败。

调用示例

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

父主题： Graph编译接口