文档
注册

框架适配接口简介

昇腾平台提供了PyTorch适配插件PT Adapter与TensorFlow适配插件TF Adapter,使得原生PyTorch/TensorFlow模型可以方便的在昇腾平台上执行训练或推理。另外,针对TensorFlow网络,昇腾平台开放了自定义Scope融合规则的能力,实现网络执行的硬件加速。

  • 若开发者网络为PyTorch 2.1.0、2.0.1、1.11.0,可参见PT Adapter API查阅PT Adapter提供的NPU自定义算子,以及NPU相对于原生PyTorch的亲合接口。
  • 若开发者网络为TensorFlow 1.15版本,可参见TF Adapter 1.x查阅TF Adapter提供的API、相关配置以及对TensorFlow 1.15 API的支持列表。
  • 若开发者网络为TensorFlow 2.6.5版本,可参见TF Adapter 2.x查阅TF Adapter提供的API、相关配置以及对TensorFlow 2.6 API的支持列表。
  • 若开发者需要自定义TensorFlow Parser Scope融合规则,可参见TF Parser Scope融合规则API查阅Ascend Graph提供的API。
搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词