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);

参数说明

1
2

typedef uint32_t(*pCallBackFunc)(uint32_t graph_id, const std::map<std::string, ge::Tensor> &params_list)；
typedef uint32_t session::(*pCallBackFunc)(uint32_t graph_id, const std::map<AscendString, ge::Tensor> &params_list);

返回值

约束说明

• 回调函数类型仅支持Summary、Save。
• 如无注册则下发summary、checkpoint算子会报错。
• 目前暂时只支持图执行完后一次性调用回调函数。