aclgrphConvertToWeightRefreshableGraphs
函数原型
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:成功。 其他值:失败。 |
调用示例
完整调用示例请参见调用示例。
父主题: Graph编译接口