TBE（Tensor Boost Engine）提供了昇騰AI處理器自定義算子開(kāi)發(fā)能力，通過(guò)TBE提供的API和自定義算子編程開(kāi)發(fā)界面可以完成相應(yīng)神經(jīng)網(wǎng)絡(luò)算子的開(kāi)發(fā)。
算子類型及名稱為TBE的重要概念：
算子類型（Type）即算子的type，代表算子的類型，例如卷積算子的類型為Convolution，在一個(gè)網(wǎng)絡(luò)中同一類型的算子可能存在多個(gè)。
算子名稱（Name）即算子的名稱，用于標(biāo)識(shí)網(wǎng)絡(luò)中的某個(gè)算子，同一網(wǎng)絡(luò)中每一個(gè)算子的名稱需要保持唯一。如下圖所示conv1，pool1，conv2都是此網(wǎng)絡(luò)中的算子名稱，其中conv1與conv2算子的類型為Convolution，表示分別做一次卷積運(yùn)算。
猜你喜歡
