华为云计算 云知识 TBE及其优势特性

TBE及其优势特性

昇腾AI软件栈提供了TBE算子开发框架,开发者可以基于此框架使用Python语言开发自定义算子。那么,我们来了解一下什么是TBE。

TBE的全称为Tensor Boost Engine,即张量加速引擎,是一款华为自研的算子开发工具,用于开发能够运行在NPU(Neural-network Processing Unit:神经网络处理器)上的TBE算子,该工具是在业界著名的开源项目TVM(Tensor Virtual Machine)基础上扩展的,提供了一套Python API来实施开发活动。

TBE的优势特性大致如下:

上一篇:人群计数及密度检测 下一篇:Predixy如何配置