框架适配接口简介
昇腾平台提供了PyTorch适配插件Ascend Extension for PyTorch与TensorFlow适配插件TF Adapter,使得原生PyTorch/TensorFlow模型可以方便的在昇腾平台上执行训练或推理。另外,针对TensorFlow网络,昇腾平台开放了自定义Scope融合规则的能力,实现网络执行的硬件加速。
- 若开发者网络为PyTorch 2.2.0、2.1.0、1.11.0,可参见《Ascend Extension for PyTorch API 参考》查阅Ascend Extension for PyTorch提供的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。