首頁/ 標簽/ leaky

如何在OneFlow中新增運算元

2開發 op在 OneFlow 中開發一個新的 user op,主要分為以下4步:定義 op實現 kernel 計算邏輯匯出 functional 介面實現用於求導的反向邏輯定義 op定義 op 指的是,對 op 的名稱,op 的輸入、輸出...

Read more