昇腾社区首页
中文
注册

前言

概述

本文档详细描述了算子基本概念,TBE算子的实现原理及实现方法,并提供了自定义算子样例供开发者参考。

如下情况下,开发者需要进行TBE自定义算子开发:
  • 昇腾AI处理器不支持开发者网络中的算子。
  • 开发者想要自己开发算子来提高计算性能。
  • 开发者想要修改现有算子中的计算逻辑。

读者对象

本文档适用于基于TBE进行算子开发的人员。

掌握以下经验和技能可以更好地理解本文档:
  • 具备Python/C++/C语言程序开发能力
  • 理解数学表达式
  • 对机器学习、深度学习有一定的了解
  • 对TVM及开源TensorFlow/Caffe/PyTorch框架有一定的了解