SaveGraphsToPb
函数功能
保存Session内图接口,将图保存为pb文件,支持Session内的权重共享功能(即Session内两张图若有两个权重值是一致的,则只会保存一份权重)。
该接口与ShardGraphsToFile接口区别:
该接口适用于任何图,而ShardGraphsToFile适用于大模型分布式编译切分场景。
函数原型
Status Session::SaveGraphsToPb(const char_t *file_path);
约束说明
同一张图内部的权重不可共享。
参数说明
参数名 |
输入/输出 |
描述 |
---|---|---|
file_path |
输入 |
保存图和权重的目录,必须为一个合法路径。 |
返回值
参数名 |
类型 |
描述 |
---|---|---|
- |
Status |
SUCCESS:成功 FAILED: 失败 |
父主题: Session类