TensorFlow网络执行流程
TensorFlow网络执行时,各组件交互流程如下图所示:
图1 交互流程


- 当用户执行训练代码后,TensorFlow前端会根据用户提供的训练脚本,生成训练模型,读取指定路径下的checkpoint文件完成模型权重初始化或随机初始化。
- 随后,框架前端会通过TF Adapter调用GE初始化接口,完成设备打开、计算引擎初始化、算子信息库初始化等操作,然后,将前端训练模型转换为IR格式的模型,并启动模型编译和执行。
- 待计算图的编译和优化都完成后,GE会调用Runtime接口分配运行资源,包含内存、 Stream、Event等,待计算资源分配完成后,交由Runtime运行时对资源进行管理。
父主题: 附录