カスタム演算子の実装と登録 - 2.5 日本語

この例では、XIR OP add を実装します。この OP は 2 つの入力テンソルを追加します。両方のテンソルは同じ形状であるものとします。

新規 XIR OP を登録する方法は、 『Vitis AI ユーザー ガイド』 (UG1414) を参照してください。この例では、add OP を含むモデルが Vitis AI xcompiler によって正しくコンパイルされており、add OP が既に Xmodel グラフに登録されているものとします。

リファレンス ゴールデン コードの詳細は、https://github.com/Xilinx/Vitis-AI/examples/Custom_OP_Demo/op_add を参照してくだい。

XIR OP を実装するには、次の手順に従います。

1. C++ クラスを作成する。
2. コンストラクター関数を作成する。
3. calculate 関数を作成する。
4. マクロを使用して実装を登録する。
5. 共有ライブラリをビルドする。
6. 運用する。