2022-05-01から1ヶ月間の記事一覧

ONNXRuntimeのカスタムオペレータを実装してみた

ソースコードはここ↓ github.com 背景とか TensorRTを使おうとしてモデルの一部をONNXのカスタムオペレータにすることがある。(TensorRTのプラグインを使うケース) ただカスタムオペレータを含むモデルはそのままではONNXRuntimeで推論できない。というこ…