RegisterCallBackFunc
函数功能
注册回调函数。
注册用户指定的summary、checkpoint回调接口。当用户下发给GE的图中带有summary、checkpoint算子时,GE会调用该回调函数。
函数原型
数据类型为string的接口后续版本会废弃,建议使用数据类型为非string的接口。
Status RegisterCallBackFunc(const std::string &key, const pCallBackFunc &callback); Status RegisterCallBackFunc(const char *key, const session::pCallBackFunc &callback);
参数说明
参数名 |
输入/输出 |
描述 |
||
---|---|---|---|---|
key |
输入 |
注册回调函数对应的关键字,字符串格式,表示回调函数类型,支持"Summary"、"Save"。 |
||
callback |
输入 |
回调函数返回的对应信息。
|
返回值
参数名 |
类型 |
描述 |
---|---|---|
- |
Status |
GE_SESSION_MANAGER_NOT_INIT:session管理未初始化。 SUCCESS:注册回调函数成功。 FAILED:注册回调函数失败。 |
约束说明
- 回调函数类型仅支持Summary、Save。
- 如无注册则下发summary、checkpoint算子会报错。
- 目前暂时只支持图执行完后一次性调用回调函数。
父主题: Graph运行接口