亮点特性
继承开源TensorFlow特性,适配昇腾AI处理器系列;实现使用昇腾AI处理器进行静态图训练,现有网络只修改数据类型和引入
FrameworkTFAdapter软件包,即可迁移到昇腾AI处理器上使用;支持自定义算子开发和模型调整
适配昇腾AI处理器
支持原生TensorFlow Python接口和拓展的Python接口
框架基础功能
支持TensorFlow图执行、图优化等功能
自定义算子开发
支持在TensorFlow框架中添加自定义算子
分布式训练
支持分布式数据并行训练,包含单机多卡、多机多卡场景
混合精度
支持混合使用FP16和FP32,基本保持精度的前提下提升性能
模型推理
输出标准的TensorFlow pb模型,通过ATC工具转换为离线推理模型