華為云計算 云知識 TBE基本概念之算子類型及名稱
TBE基本概念之算子類型及名稱

TBE(Tensor Boost Engine)提供了昇騰AI處理器自定義算子開發(fā)能力,通過TBE提供的API和自定義算子編程開發(fā)界面可以完成相應神經網絡算子的開發(fā)。

算子類型及名稱為TBE的重要概念:

算子類型(Type)即算子的type,代表算子的類型,例如卷積算子的類型為Convolution,在一個網絡中同一類型的算子可能存在多個。

算子名稱(Name)即算子的名稱,用于標識網絡中的某個算子,同一網絡中每一個算子的名稱需要保持唯一。如下圖所示conv1,pool1,conv2都是此網絡中的算子名稱,其中conv1與conv2算子的類型為Convolution,表示分別做一次卷積運算。