前言 概述 本文档详细描述了算子基本概念,TBE算子的实现原理及实现方法,并提供了自定义算子样例供开发者参考。 如下情况下,开发者需要进行TBE自定义算子开发: 昇腾AI处理器不支持开发者网络中的算子。 开发者想要自己开发算子来提高计算性能。 开发者想要修改现有算子中的计算逻辑。 读者对象 本文档适用于基于TBE进行算子开发的人员。 掌握以下经验和技能可以更好地理解本文档: 具备Python/C++/C语言程序开发能力 理解数学表达式 对机器学习、深度学习有一定的了解 对TVM及开源TensorFlow/Caffe/PyTorch框架有一定的了解