迁移思路
TF Adapter适配TensorFlow 1.15的思路是:
利用TensorFlow提供的config扩展机制,将NPU上的相关功能配置通过config的方式向下传递;然后再利用TensorFlow提供的优化器注册机制注册NPU上的优化器并根据config配置对图进行处理;最终将处理好的图下发到CANN平台执行。
所以手工迁移的主要点是添加NPU上的config配置,将扩展后的NPU config传入tf.Session的config参数。TensorFlow 1.15中常见的Estimator、sess.run、Keras三种脚本最终都是调用sess.run实现的,所以三种脚本的迁移方式本质上是一致的。
父主题: 手工迁移