昇騰AI處理器軟件棧--張量加速引擎(TBE)

神經網絡構造中,算子組成了不同應用功能的網絡結構。而張量加速引擎(Tensor Boost Engine)作爲算子的兵工廠,爲基於昇騰AI處理器運行的神經網絡提供算子開發能力,用TBE語言編寫的TBE算子來構建各種神經網絡模型。同時,TBE對算子也提供了封裝調用能力。在TBE中有一個優化過的神經網絡TBE標準算子庫,開發者可以直接利用標準算子庫中的算子實現高性能的神經網絡計算。除此之外,TBE也提
相關文章
相關標籤/搜索