華為云計算 云知識 TBE及其優(yōu)勢特性
TBE及其優(yōu)勢特性

昇騰AI軟件棧提供了TBE算子開發(fā)框架,開發(fā)者可以基于此框架使用Python語言開發(fā)自定義算子。那么,我們來了解一下什么是TBE。

TBE的全稱為Tensor Boost Engine,即張量加速引擎,是一款華為自研的算子開發(fā)工具,用于開發(fā)能夠運行在NPU(Neural-network Processing Unit:神經(jīng)網(wǎng)絡(luò)處理器)上的TBE算子,該工具是在業(yè)界著名的開源項目TVM(Tensor Virtual Machine)基礎(chǔ)上擴(kuò)展的,提供了一套Python API來實施開發(fā)活動。

TBE的優(yōu)勢特性大致如下: