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