- 头文件:#include <ge/ge_api_v2.h>
- 库文件:libge_runner_v2.so
注册回调函数。
注册用户指定的Summary、Checkpoint回调接口。当用户下发给GE的图中带有Summary、Checkpoint算子时,GE会调用该回调函数。
[object Object]
[object Object]
- 回调函数类型仅支持Summary、Save。
- 如无注册则下发Summary、Checkpoint算子会报错。
- 目前暂时只支持图执行完后一次性调用回调函数。
注册回调函数。
注册用户指定的Summary、Checkpoint回调接口。当用户下发给GE的图中带有Summary、Checkpoint算子时,GE会调用该回调函数。