2021-08-01から1ヶ月間の記事一覧

PyTorchモデルをONNX化する際にカスタムオペレータの実装が必要っぽい件

やりたかったこと PyTorchモデルをONNXファイルにexportしたい モデルにはカスタムオペレータを含む カスタムオペレータの実装は存在しない(該当部分は推論不能) カスタムオペレータはONNXでそれらしきop_typeを持つノードで出力してくれればOK やってみた…