56 Star 128 Fork 255

Ascend/tensorflow

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
克隆/下载
贡献代码
同步代码
取消
提示: 由于 Git 不支持空文件夾,创建文件夹后会生成空的 .keep 文件
Loading...
README
Apache-2.0

TensorFlow Adapter For Ascend

简介

TensorFlow Adapter For Ascend(简称TF Adapter)是昇腾提供的TensorFlow框架适配插件,让Tensorflow框架的开发者可以使用昇腾AI处理器的算力。 开发者只需安装TF Adapter插件,并在现有TensorFlow脚本中添加少量配置,即可实现在昇腾AI处理器上加速自己的训练任务。

版本配套说明

CANN版本配套说明

TF Adapter源码标签的命名规则为:tfa_{标签名}_{版本号},其中{版本号}与配套的CANN版本号保持一致。

需要注意:

使用TF Adapter时,请选择与配套CANN版本一致的标签源码,使用master分支可能存在版本不匹配的风险。

支持的TensorFlow版本

TF Adapter支持的TensorFlow版本为TensorFlow 1.15与TensorFlow 2.6.5。

支持的产品型号

Atlas 训练系列产品

Atlas A2 训练系列产品

Atlas 推理系列产品(Ascend 310P处理器)(仅支持TensorFlow 1.15在线推理特性)

如何使用源码

若您的TensorFlow框架版本是1.15,本源码仓的编译安装等详细使用方法请参见tf_adapter 1.x

若您的TensorFlow框架版本是2.6.5,本源码仓的编译安装等详细使用方法请参见tf_adapter 2.x

贡献指南

针对tensorflow仓,开发者准备本地代码与提交PR时需要重点关注如下几点:

  1. 提交PR时,请按照PR模板仔细填写本次PR的业务背景、目的、方案等信息。
  2. 若您的修改不是简单的bug修复,而是涉及到新增特性、新增接口、新增配置参数或者修改代码流程等,请务必先通过Issue进行方案讨论,以避免您的代码被拒绝合入。若您不确定本次修改是否可被归为“简单的bug修复”,亦可通过提交Issue进行方案讨论。
  3. psuh代码前,请务必保证已经完成了基础功能测试和网络测试。

许可证

Apache License 2.0

空文件

简介

Ascend TensorFlow Adapter 展开 收起
README
Apache-2.0
取消

发行版 (2)

全部

贡献者

全部

近期动态

不能加载更多了
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
C++
1
https://gitee.com/ascend/tensorflow.git
git@gitee.com:ascend/tensorflow.git
ascend
tensorflow
tensorflow
master

搜索帮助