昇腾社区首页
中文
注册

SaveGraphsToPb

产品支持情况

产品

是否支持

Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件

Atlas 200I/500 A2 推理产品

x

Atlas 推理系列产品

Atlas 训练系列产品

Atlas 200/300/500 推理产品

x

功能说明

保存Session内图接口,将图保存为pb文件,支持Session内的权重共享功能(即Session内两张图若有两个权重值是一致的,则只会保存一份权重)。

该接口与ShardGraphsToFile接口区别:

该接口适用于任何图,而ShardGraphsToFile适用于大模型分布式编译切分场景。

函数原型

1
Status SaveGraphsToPb(const char_t *file_path) const

参数说明

参数名

输入/输出

说明

file_path

输入

保存图和权重的目录,必须为一个合法路径。

返回值说明

参数名

类型

说明

-

Status

SUCCESS:成功

FAILED: 失败

约束说明

同一张图内部的权重不可共享。