开发者
下载
[object Object][object Object]
  • 头文件:#include <ge/ge_api.h>
  • 库文件:libge_runner.so
[object Object]

注册回调函数。

注册用户指定的Summary、Checkpoint回调接口。当用户下发给GE的图中带有Summary、Checkpoint算子时,GE会调用该回调函数。

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