文档
注册

SaveGraphsToPb

函数功能

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

该接口与ShardGraphsToFile接口区别:

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

函数原型

Status Session::SaveGraphsToPb(const char_t *file_path);

约束说明

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

参数说明

参数名

输入/输出

描述

file_path

输入

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

返回值

参数名

类型

描述

-

Status

SUCCESS:成功

FAILED: 失败

搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词